Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream.h334 const dnn::AlgorithmConfig &algorithm_config, in ConvolveWithAlgorithm()
445 const dnn::AlgorithmConfig &algorithm_config, in ConvolveBackwardDataWithAlgorithm()
477 const dnn::AlgorithmConfig &algorithm_config, in ConvolveBackwardFilterWithAlgorithm()
Ddnn.h1154 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
1172 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
1193 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
1212 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
1250 const AlgorithmConfig& algorithm_config, in PrepareForConvolution()
2115 const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2685 const AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
Dstream_executor_pimpl.cc369 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config, in createRnnDescriptor()
Dstream.cc457 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
488 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
520 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
551 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
582 const dnn::AlgorithmConfig &algorithm_config, in FusedConvolveWithAlgorithm()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1065 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor()
1093 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in Create()
1212 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config() function in stream_executor::gpu::CudnnRnnDescriptor
2074 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2689 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionForwardAlgorithm()
2752 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardDataAlgorithm()
2813 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardFilterAlgorithm()
2990 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3189 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolveImpl()
3767 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.cc388 AlgorithmConfig algorithm_config; in operator ()() local
Dconv_ops_3d.cc505 AlgorithmConfig algorithm_config; in launch() local
Dconv_grad_ops_3d.cc1424 AlgorithmConfig algorithm_config; in Compute() local
1921 AlgorithmConfig algorithm_config; in Compute() local
Dconv_grad_filter_ops.cc983 AlgorithmConfig algorithm_config; in operator ()() local
Dconv_ops.cc985 AlgorithmConfig algorithm_config; in operator ()() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2519 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2850 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3686 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3702 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3719 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3735 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()