| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_thunk.cc | 119 se::blas::AlgorithmType algorithm, se::Stream* stream, in DoGemmWithAlgorithm()
157 StatusOr<se::blas::AlgorithmType> DoGemmAutotune( in DoGemmAutotune()
161 std::vector<se::blas::AlgorithmType> algorithms; in DoGemmAutotune()
434 StatusOr<se::blas::AlgorithmType> best_algorithm = GetGemmAutotuneFn( in ExecuteOnStream()
450 const StatusOr<se::blas::AlgorithmType>& best_algorithm = in ExecuteOnStream()
|
| D | gemm_thunk.h | 84 std::unordered_map<string, StatusOr<se::blas::AlgorithmType>>
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 112 typedef int64 AlgorithmType; typedef
113 constexpr AlgorithmType kDefaultAlgorithm = -1;
114 constexpr AlgorithmType kDefaultBlasGemm = -2;
115 constexpr AlgorithmType kDefaultBlasGemv = -3;
116 constexpr AlgorithmType kNoAlgorithm = -4;
125 constexpr AlgorithmType kDefaultGemmAlgo = -1;
136 AlgorithmType algorithm() const { return algorithm_; } in algorithm()
137 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm()
143 AlgorithmType algorithm_ = kDefaultAlgorithm;
150 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig()
[all …]
|
| D | stream.h | 1330 blas::AlgorithmType algorithm,
1338 blas::AlgorithmType algorithm,
1346 blas::AlgorithmType algorithm,
1354 blas::AlgorithmType algorithm,
1363 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
1372 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
|
| D | stream_executor_pimpl.h | 403 bool GetBlasGemmAlgorithms(std::vector<blas::AlgorithmType> *out_algorithms);
|
| D | stream.cc | 3793 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
3804 int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3816 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3827 DeviceMemory<int> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3840 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
3850 DeviceMemory<float> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3863 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
3873 DeviceMemory<double> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3888 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3902 blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
[all …]
|
| D | stream_executor_pimpl.cc | 372 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | matmul_op.cc | 115 typedef se::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/rocm/ |
| D | rocm_blas.h | 123 blas::AlgorithmType algorithm,
|
| D | rocm_blas.cc | 1703 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithmImpl()
1709 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
1719 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
1734 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
1747 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
1760 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
1774 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
1789 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 125 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
|
| D | cuda_blas.cc | 1862 static bool UsesTensorOps(blas::AlgorithmType algo) { in UsesTensorOps()
1892 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithmImpl()
1982 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
2045 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
2059 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
2087 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
2099 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
2112 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
2126 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
|
| /external/python/cpython2/Modules/ |
| D | clmodule.c | 903 func(AlgorithmType, Int_FromLong) in func()
|