| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | matmul_op_fused.cc | 191 se::blas::AlgorithmConfig algorithm_config(se::blas::kNoAlgorithm); in AutotuneMatmul() local
193 &algorithm_config)) { in AutotuneMatmul()
224 algorithm_config.set_algorithm(best_result.algorithm()); in AutotuneMatmul()
229 AutoTuneBatchMatmul::GetInstance()->Insert(matmul_params, algorithm_config); in AutotuneMatmul()
231 return algorithm_config; in AutotuneMatmul()
311 se::blas::AlgorithmConfig algorithm_config = in operator ()() local
314 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm(); in operator ()()
|
| H A D | matmul_op_impl.h | 443 se::blas::AlgorithmConfig algorithm_config(se::blas::kNoAlgorithm);
445 algorithm_config.set_algorithm(0);
447 matmul_params, &algorithm_config)) {
478 algorithm_config.set_algorithm(best_result.algorithm());
484 algorithm_config);
486 se::blas::AlgorithmType algorithm_idx = algorithm_config.algorithm();
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 1240 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor() argument
1250 rnn_algo_(ToCudnnRNNAlgo(algorithm_config.algorithm())), in CudnnRnnDescriptor()
1256 algorithm_config_(algorithm_config), in CudnnRnnDescriptor()
1268 const dnn::AlgorithmConfig& algorithm_config, float dropout, in Create() argument
1275 cudnnRNNAlgo_t rnn_algo = ToCudnnRNNAlgo(algorithm_config.algorithm()); in Create()
1293 algorithm_config.algorithm().has_value() in Create()
1294 ? algorithm_config.algorithm()->tensor_ops_enabled() in Create()
1375 data_type, compute_type, algorithm_config, in Create()
1390 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config() function in stream_executor::gpu::CudnnRnnDescriptor
2022 auto algo_desc = *rnn_desc.algorithm_config().algorithm(); in DoRnnForwardImpl()
[all …]
|
| 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,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 1313 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
1329 const AlgorithmConfig& algorithm_config, in PrepareForConvolution() argument
1335 output_data, convolution_descriptor, algorithm_config, in PrepareForConvolution()
2043 const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor() argument
2625 const AlgorithmConfig& algorithm_config, in DoPrepareForConvolution() argument
|
| H A D | stream.h | 345 const dnn::AlgorithmConfig &algorithm_config, in ConvolveWithAlgorithm() argument
353 algorithm_config, scratch_allocator, &algorithm_desc, in ConvolveWithAlgorithm()
378 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm() argument
388 scratch_allocator, algorithm_config, output_profile_result); in FusedConvolveWithAlgorithm()
|
| H A D | stream_executor_pimpl.cc | 379 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor() argument
389 direction_mode, rnn_mode, data_type, algorithm_config, dropout, seed, in createRnnDescriptor()
|
| H A D | stream_executor_pimpl.h | 423 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config,
|
| /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,
|
| H A D | rocm_dnn.cc | 2573 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor() argument
2910 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution() argument
2914 algorithm_config.algorithm(); in DoPrepareForConvolution()
2921 assert(algorithm_config.scratch_size().has_value()); in DoPrepareForConvolution()
2923 size_t scratch_memory_size = *(algorithm_config.scratch_size()); in DoPrepareForConvolution()
3808 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve() argument
|
| /aosp_15_r20/external/googleapis/google/cloud/aiplatform/v1beta1/ |
| H A D | feature_view.proto | 97 oneof algorithm_config { oneof
176 oneof algorithm_config { oneof
|
| /aosp_15_r20/external/googleapis/google/cloud/aiplatform/v1/ |
| H A D | feature_view.proto | 96 oneof algorithm_config { oneof
|