| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gemm_algorithm_picker.cc | 50 absl::optional<se::blas::AlgorithmType>>();
60 static StatusOr<absl::optional<se::blas::AlgorithmType>> DoUncachedGemmAutotune( in DoUncachedGemmAutotune()
112 std::vector<se::blas::AlgorithmType> algorithms; in DoUncachedGemmAutotune()
115 absl::optional<se::blas::AlgorithmType> first_algorithm; in DoUncachedGemmAutotune()
120 for (se::blas::AlgorithmType algorithm : algorithms) { in DoUncachedGemmAutotune()
231 static StatusOr<absl::optional<se::blas::AlgorithmType>> DoGemmAutotune( in DoGemmAutotune()
263 absl::optional<se::blas::AlgorithmType> result; in DoGemmAutotune()
289 TF_ASSIGN_OR_RETURN(absl::optional<se::blas::AlgorithmType> gemm_algorithm, in RunOnInstruction()
|
| D | gemm_thunk.cc | 85 se::Stream *stream, absl::optional<se::blas::AlgorithmType> algorithm, in DoGemmWithAlgorithm()
178 absl::optional<se::blas::AlgorithmType> algorithm) { in RunGemm()
259 auto best_algorithm = [&]() -> absl::optional<se::blas::AlgorithmType> { in RunGemm()
|
| D | gemm_thunk.h | 86 absl::optional<se::blas::AlgorithmType> algorithm = absl::nullopt);
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | blas.h | 143 typedef int64 AlgorithmType; typedef
144 constexpr AlgorithmType kDefaultAlgorithm = -1;
145 constexpr AlgorithmType kDefaultBlasGemm = -2;
146 constexpr AlgorithmType kDefaultBlasGemv = -3;
147 constexpr AlgorithmType kNoAlgorithm = -4;
156 constexpr AlgorithmType kDefaultGemmAlgo = -1;
167 AlgorithmType algorithm() const { return algorithm_; } in algorithm()
168 void set_algorithm(AlgorithmType val) { algorithm_ = val; } in set_algorithm()
174 AlgorithmType algorithm_ = kDefaultAlgorithm;
181 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig()
[all …]
|
| D | stream.h | 1353 blas::AlgorithmType algorithm,
1361 blas::AlgorithmType algorithm,
1369 blas::AlgorithmType algorithm,
1377 blas::AlgorithmType algorithm,
1386 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
1395 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
|
| D | stream_executor_pimpl.h | 396 bool GetBlasGemmAlgorithms(std::vector<blas::AlgorithmType> *out_algorithms);
|
| D | stream.cc | 3436 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
3447 int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3459 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3470 DeviceMemory<int> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3483 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
3493 DeviceMemory<float> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3506 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in ThenBlasGemmWithAlgorithm()
3516 DeviceMemory<double> *, int, blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
3531 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in ThenBlasGemmWithAlgorithm()
3545 blas::ComputationType, blas::AlgorithmType> in ThenBlasGemmWithAlgorithm()
[all …]
|
| D | stream_executor_pimpl.cc | 332 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/stream_executor/cuda/ |
| D | cuda_blas.h | 122 blas::ComputationType computation_type, blas::AlgorithmType algorithm,
|
| D | cuda_blas.cc | 2057 static bool UsesTensorOps(blas::AlgorithmType algo) { in DoBlasInternalImpl()
2072 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl()
2193 std::vector<blas::AlgorithmType> *out_algorithms) { in DoBlasInternalImpl()
2256 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl()
2270 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
2298 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
2310 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasInternalImpl()
2323 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl()
2337 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasInternalImpl()
3369 CUDABlasLtMatmulAlgorithm(blas::AlgorithmType index, in DoBlasInternalImpl()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.h | 174 blas::AlgorithmType algorithm,
|
| D | rocm_blas.cc | 1838 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithmImpl()
1844 std::vector<blas::AlgorithmType> *out_algorithms) { in GetBlasGemmAlgorithms()
1854 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
1869 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
1882 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
1895 blas::AlgorithmType algorithm, blas::ProfileResult *output_profile_result) { in DoBlasGemmWithAlgorithm()
1909 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
1924 blas::ComputationType computation_type, blas::AlgorithmType algorithm, in DoBlasGemmWithAlgorithm()
|
| /external/python/cpython2/Modules/ |
| D | clmodule.c | 903 func(AlgorithmType, Int_FromLong) in func()
|