Searched refs:AlgorithmType (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 94 se::blas::AlgorithmType algorithm, se::Stream* stream, in DoGemmWithAlgorithm()
126 StatusOr<se::blas::AlgorithmType> DoGemmAutotune( in DoGemmAutotune()
130 std::vector<se::blas::AlgorithmType> algorithms; in DoGemmAutotune()
291 StatusOr<se::blas::AlgorithmType> best_algorithm = in ExecuteOnStream()
307 const StatusOr<se::blas::AlgorithmType>& best_algorithm = in ExecuteOnStream()
|
| D | gemm_thunk.h | 81 StatusOr<::perftools::gputools::blas::AlgorithmType>>
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 109 typedef int64 AlgorithmType; typedef
110 constexpr AlgorithmType kDefaultAlgorithm = -1;
111 constexpr AlgorithmType kDefaultBlasGemm = -2;
112 constexpr AlgorithmType kDefaultBlasGemv = -3;
113 constexpr AlgorithmType kNoAlgorithm = -4;
122 constexpr AlgorithmType kDefaultGemmAlgo = -1;
133 AlgorithmType algorithm() const { return algorithm_; } in algorithm()
134 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm()
140 AlgorithmType algorithm_ = kDefaultAlgorithm;
147 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig()
[all …]
|
| D | stream.h | 1394 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
1403 blas::AlgorithmType algorithm,
1412 blas::AlgorithmType algorithm,
1419 blas::AlgorithmType algorithm,
1427 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
1435 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
|
| D | stream_executor_pimpl.h | 368 bool GetBlasGemmAlgorithms(std::vector<blas::AlgorithmType> *out_algorithms);
|
| D | stream.cc | 3789 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3801 blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3813 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
3822 DeviceMemory<int> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3834 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
3843 DeviceMemory<float> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3855 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
3865 blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3878 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3890 blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
[all …]
|
| D | stream_executor_pimpl.cc | 335 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.cc | 115 typedef perftools::gputools::blas::AlgorithmType AlgorithmType; typedef
117 typedef int64 AlgorithmType;
123 std::vector<AlgorithmType>* algorithms, bool use_aututone, Tensor* out) { in launch()
|
| /external/python/cpython2/Lib/plat-irix5/ |
| D | CL_old.py | 104 def AlgorithmType(scheme): function
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 131 blas::AlgorithmType algorithm,
|
| D | cuda_blas.cc | 2130 static bool UsesTensorOps(blas::AlgorithmType algo) { in UsesTensorOps()
2156 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithmImpl()
2214 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
2245 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
2257 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
2269 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
2280 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
2292 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
2305 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
|
| /external/python/cpython2/Modules/ |
| D | clmodule.c | 903 func(AlgorithmType, Int_FromLong) in func()
|