| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | cudnn_rnn_ops.cc | 127 using se::dnn::AlgorithmConfig;
229 AutotuneSingleton<RnnAutotuneGroup, CudnnRnnParameters, AlgorithmConfig>;
1082 const AlgorithmConfig algo_config; in ExtractCudnnRNNParamsInfo()
1099 const AlgorithmConfig& algo_config, in CreateRnnDescriptor()
1126 const AlgorithmConfig& algo_config, in GetCachedRnnDescriptor()
1488 AlgorithmConfig algo_config; in Compute()
1495 AlgorithmConfig* output_algo_config, in ComputeAndReturnAlgorithm()
1575 AlgorithmConfig* best_algo_config) { in MaybeAutotune()
1577 *best_algo_config = AlgorithmConfig(); in MaybeAutotune()
1643 AlgorithmConfig best_algo_config; in Compute()
[all …]
|
| H A D | gpu_utils.h | 291 explicit AutotuneEntry(se::dnn::AlgorithmConfig config) in AutotuneEntry()
353 const se::dnn::AlgorithmConfig& GetAlgorithmConfig() const { in GetAlgorithmConfig()
394 se::dnn::AlgorithmConfig algorithm_config_;
406 StatusOr<se::dnn::AlgorithmConfig> BestCudnnConvAlgorithm(
|
| H A D | matmul_op_fused.cc | 184 se::blas::AlgorithmConfig AutotuneMatmul( in AutotuneMatmul()
191 se::blas::AlgorithmConfig algorithm_config(se::blas::kNoAlgorithm); in AutotuneMatmul()
311 se::blas::AlgorithmConfig algorithm_config = in operator ()()
|
| H A D | gpu_utils.cc | 271 StatusOr<se::dnn::AlgorithmConfig> BestCudnnConvAlgorithm( in BestCudnnConvAlgorithm()
282 se::dnn::AlgorithmConfig result( in BestCudnnConvAlgorithm()
|
| H A D | matmul_op_impl.h | 295 se::blas::AlgorithmConfig,
443 se::blas::AlgorithmConfig algorithm_config(se::blas::kNoAlgorithm);
|
| H A D | conv_ops_gpu.cc | 414 se::dnn::AlgorithmConfig(profile_algorithm, in AutotuneUnfusedConv()
|
| H A D | conv_grad_input_ops.cc | 79 using se::dnn::AlgorithmConfig; in operator ()()
|
| H A D | conv_ops_3d.cc | 502 using se::dnn::AlgorithmConfig; in launch()
|
| H A D | conv_grad_ops_3d.cc | 1517 using se::dnn::AlgorithmConfig; in Compute()
1916 using se::dnn::AlgorithmConfig; in Compute()
|
| H A D | conv_grad_filter_ops.cc | 675 using se::dnn::AlgorithmConfig; in operator ()()
|
| H A D | conv_ops.cc | 807 using se::dnn::AlgorithmConfig; in operator ()()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 924 class AlgorithmConfig {
926 AlgorithmConfig() {} in AlgorithmConfig() function
927 explicit AlgorithmConfig(AlgorithmDesc algorithm) : algorithm_(algorithm) {} in AlgorithmConfig() function
928 AlgorithmConfig(AlgorithmDesc algorithm, size_t scratch_size) in AlgorithmConfig() function
930 AlgorithmConfig(AlgorithmDesc algorithm, AlgorithmDesc algorithm_no_scratch) in AlgorithmConfig() function
932 AlgorithmConfig(AlgorithmDesc algorithm, size_t scratch_size, in AlgorithmConfig() function
940 explicit AlgorithmConfig(const AlgorithmConfigProto& algorithm_config_proto) { in AlgorithmConfig() function
965 bool operator==(const AlgorithmConfig& other) const {
970 bool operator!=(const AlgorithmConfig& other) const {
1313 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
[all …]
|
| H A D | blas.h | 167 class AlgorithmConfig {
169 AlgorithmConfig() : algorithm_(kDefaultAlgorithm) {} in AlgorithmConfig() function
170 explicit AlgorithmConfig(AlgorithmType algorithm) : algorithm_(algorithm) {} in AlgorithmConfig() function
173 bool operator==(const AlgorithmConfig &other) const {
176 bool operator!=(const AlgorithmConfig &other) const {
|
| H A D | blas.cc | 71 std::string AlgorithmConfig::ToString() const { in ToString()
|
| H A D | dnn.proto | 137 // Proto definition of AlgorithmConfig in "dnn.h".
|
| H A D | stream_executor_pimpl.h | 423 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config,
|
| H A D | stream.h | 345 const dnn::AlgorithmConfig &algorithm_config, in ConvolveWithAlgorithm()
378 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
|
| H A D | stream_executor_pimpl.cc | 379 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
|
| H A D | dnn.cc | 447 std::string AlgorithmConfig::ToString() const { in ToString()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/autotune_maps/ |
| H A D | autotune_serialize_test.cc | 32 using stream_executor::dnn::AlgorithmConfig;
159 AlgorithmConfig algorithm_config_example_a(algorithm, /*scratch_size=*/1, in TEST()
|
| H A D | autotune_serialize.cc | 38 using stream_executor::dnn::AlgorithmConfig;
153 entry = AutotuneEntry<Op>(AlgorithmConfig(algorithm_config_proto)); in PopulateConvMap()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.h | 61 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config,
360 const dnn::AlgorithmConfig& algorithm_config,
631 const dnn::AlgorithmConfig& algorithm_config,
|
| H A D | cuda_dnn.cc | 1240 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor()
1268 const dnn::AlgorithmConfig& algorithm_config, float dropout, in Create()
1390 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config()
1424 dnn::AlgorithmConfig algorithm_config_;
2417 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
3054 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionForwardAlgorithm()
3117 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardDataAlgorithm()
3178 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardFilterAlgorithm()
3963 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
5766 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.h | 88 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config,
364 const dnn::AlgorithmConfig& algorithm_config,
769 const dnn::AlgorithmConfig& algorithm_config,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_conv_runner.cc | 60 using se::dnn::AlgorithmConfig;
|