Searched refs:AlgorithmType (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| 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()
|
| D | matmul_utils.h | 117 std::optional<se::blas::AlgorithmType> algorithm = std::nullopt,
|
| 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()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| 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 …]
|
| 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()
|
| D | stream_executor_pimpl.h | 415 std::vector<blas::AlgorithmType>* out_algorithms);
|
| D | stream_executor_pimpl.cc | 366 Stream* stream, std::vector<blas::AlgorithmType>* out_algorithms) { in GetBlasGemmAlgorithms()
|
| /external/python/cpython2/Lib/plat-irix5/ |
| D | CL_old.py | 104 def AlgorithmType(scheme): function
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| 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()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op_fused.cc | 314 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm(); in operator ()()
|
| D | matmul_op_impl.h | 486 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm();
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| 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()
|
| /external/python/cpython2/Modules/ |
| D | clmodule.c | 903 func(AlgorithmType, Int_FromLong) in func()
|