| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | blas.h | 132 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 D | stream.h | 870 blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
885 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
924 blas::AlgorithmType algorithm, in ThenBlasGemmStridedBatchedWithAlgorithm()
|
| H A D | stream_executor_pimpl.h | 415 std::vector<blas::AlgorithmType>* out_algorithms);
|
| H A D | stream_executor_pimpl.cc | 366 Stream* stream, std::vector<blas::AlgorithmType>* out_algorithms) { in GetBlasGemmAlgorithms()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gemm_algorithm_picker.cc | 209 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 D | matmul_utils.cc | 428 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 D | matmul_utils.h | 117 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/ |
| D | certenroll.h | 34 typedef enum AlgorithmType { enum
42 } AlgorithmType; typedef
|
| D | bdaiface.h | 6240 DWORD *AlgorithmType,
6248 DWORD *AlgorithmType,
6315 DWORD *AlgorithmType,
6324 DWORD *AlgorithmType,
6377 …tIndex,MethodID,AlgorithmType,SamplingRate,BitDepth,NumChannels) (This)->lpVtbl->EnumAudioCapabili…
6378 …AlgorithmType,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 D | cuda_blas.cc | 911 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/ |
| D | CL_old.py | 104 def AlgorithmType(scheme): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_blas.cc | 635 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 D | matmul_op_fused.cc | 314 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm(); in operator ()()
|
| H A D | matmul_op_impl.h | 486 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm();
|
| /aosp_15_r20/external/python/cpython2/Modules/ |
| D | clmodule.c | 903 func(AlgorithmType, Int_FromLong) in func()
|