Home
last modified time | relevance | path

Searched refs:AlgorithmType (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dblas.h132 typedef int64_t AlgorithmType; typedef
133 constexpr AlgorithmType kDefaultAlgorithm = -1;
134 constexpr AlgorithmType kDefaultBlasGemm = -2;
135 constexpr AlgorithmType kDefaultBlasGemv = -3;
136 constexpr AlgorithmType kNoAlgorithm = -4;
145 constexpr AlgorithmType kDefaultGemmAlgo = -1;
156 AlgorithmType algorithm() const { return algorithm_; } in algorithm()
157 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm()
163 AlgorithmType algorithm_ = kDefaultAlgorithm;
170 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig()
[all …]
H A Dstream.h870 blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
885 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
924 blas::AlgorithmType algorithm, in ThenBlasGemmStridedBatchedWithAlgorithm()
H A Dstream_executor_pimpl.h415 std::vector<blas::AlgorithmType>* out_algorithms);
H A Dstream_executor_pimpl.cc366 Stream* stream, std::vector<blas::AlgorithmType>* out_algorithms) { in GetBlasGemmAlgorithms()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgemm_algorithm_picker.cc209 StatusOr<std::optional<se::blas::AlgorithmType>> DoGemmAutotune( in DoGemmAutotune()
227 std::optional<se::blas::AlgorithmType>>(); in DoGemmAutotune()
267 std::optional<se::blas::AlgorithmType> best_algorithm; in DoGemmAutotune()
308 std::vector<se::blas::AlgorithmType> algorithms; in DoGemmAutotune()
312 GetBestAlgorithm<se::blas::AlgorithmType>( in DoGemmAutotune()
315 [&](const se::blas::AlgorithmType& algorithm) in DoGemmAutotune()
350 TF_ASSIGN_OR_RETURN(std::optional<se::blas::AlgorithmType> gemm_algorithm, in RunOnInstruction()
H A Dmatmul_utils.cc428 se::blas::AlgorithmType algorithm, in DoGemmWithAlgorithm()
458 std::optional<se::blas::AlgorithmType> algorithm, in DoGemm()
489 std::optional<se::blas::AlgorithmType> algorithm, in RunGemm()
H A Dmatmul_utils.h117 std::optional<se::blas::AlgorithmType> algorithm = std::nullopt,
/aosp_15_r20/prebuilts/gcc/linux-x86/host/x86_64-w64-mingw32-4.8/x86_64-w64-mingw32/include/
Dcertenroll.h34 typedef enum AlgorithmType { enum
42 } AlgorithmType; typedef
Dbdaiface.h6240 DWORD *AlgorithmType,
6248 DWORD *AlgorithmType,
6315 DWORD *AlgorithmType,
6324 DWORD *AlgorithmType,
6377 …tIndex,MethodID,AlgorithmType,SamplingRate,BitDepth,NumChannels) (This)->lpVtbl->EnumAudioCapabili…
6378AlgorithmType,VerticalSize,HorizontalSize,AspectRatio,FrameRateCode,ProgressiveSequence) (This)->l…
6396 …ioCapability(IBDA_Encoder* This,DWORD FmtIndex,DWORD *MethodID,DWORD *AlgorithmType,DWORD *Samplin…
6397 …return This->lpVtbl->EnumAudioCapability(This,FmtIndex,MethodID,AlgorithmType,SamplingRate,BitDept…
6399 …eoCapability(IBDA_Encoder* This,DWORD FmtIndex,DWORD *MethodID,DWORD *AlgorithmType,DWORD *Vertica…
6400 …return This->lpVtbl->EnumVideoCapability(This,FmtIndex,MethodID,AlgorithmType,VerticalSize,Horizon…
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas.cc911 static bool UsesTensorOps(blas::AlgorithmType algo) { in DoBlasInternalImpl()
921 Stream *stream, blas::AlgorithmType algorithm, blas::DataType type_a, in DoBlasInternalImpl()
999 GpuTimer *timer, blas::AlgorithmType algorithm, in DoBlasInternalImpl()
1021 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1051 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
1098 Stream *stream, std::vector<blas::AlgorithmType> *out_algorithms) { in DoBlasInternalImpl()
/aosp_15_r20/external/python/cpython2/Lib/plat-irix5/
DCL_old.py104 def AlgorithmType(scheme): function
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_blas.cc635 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
647 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmStridedBatchedWithAlgorithm()
653 Stream *stream, std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_op_fused.cc314 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm(); in operator ()()
H A Dmatmul_op_impl.h486 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm();
/aosp_15_r20/external/python/cpython2/Modules/
Dclmodule.c903 func(AlgorithmType, Int_FromLong) in func()