Home
last modified time | relevance | path

Searched defs:algorithm_config (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/googleapis/google/cloud/aiplatform/v1beta1/
H A Dfeature_view.proto97 oneof algorithm_config { oneof
176 oneof algorithm_config { oneof
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_op_fused.cc191 se::blas::AlgorithmConfig algorithm_config(se::blas::kNoAlgorithm); in AutotuneMatmul() local
311 se::blas::AlgorithmConfig algorithm_config = in operator ()() local
/aosp_15_r20/external/googleapis/google/cloud/aiplatform/v1/
H A Dfeature_view.proto96 oneof algorithm_config { oneof
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dstream.h345 const dnn::AlgorithmConfig &algorithm_config, in ConvolveWithAlgorithm()
378 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
H A Ddnn.h1313 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
1329 const AlgorithmConfig& algorithm_config, in PrepareForConvolution()
2043 const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2625 const AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
H A Dstream_executor_pimpl.cc379 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc1240 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() function in stream_executor::gpu::CudnnRnnDescriptor
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 Drocm_dnn.cc2573 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2910 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3808 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()