Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1048 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor()
1076 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in Create()
1183 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config() function in stream_executor::gpu::CudnnRnnDescriptor
2071 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2599 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionForwardAlgorithm()
2652 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardDataAlgorithm()
2704 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardFilterAlgorithm()
2900 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3225 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolveImpl()
3789 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc446 AlgorithmConfig algorithm_config; in launch() local
Dmatmul_op.cc294 AlgorithmConfig algorithm_config(kNoAlgorithm); in launch() local
Dconv_grad_ops_3d.cc1371 AlgorithmConfig algorithm_config; in Compute() local
1860 AlgorithmConfig algorithm_config; in Compute() local
Dconv_grad_filter_ops.cc978 AlgorithmConfig algorithm_config; in operator ()() local
Dconv_ops.cc985 AlgorithmConfig algorithm_config; in operator ()() local
Dconv_grad_input_ops.cc1146 AlgorithmConfig algorithm_config; in operator ()() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc471 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
509 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
548 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
586 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
624 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
659 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
701 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
743 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
785 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
827 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
[all …]
Dstream_executor_pimpl.cc358 dnn::DataType data_type, const dnn::AlgorithmConfig &algorithm_config, in createRnnDescriptor()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2323 dnn::DataType data_type, const dnn::AlgorithmConfig& algorithm_config, in createRnnDescriptor()
2643 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3445 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3462 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3480 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3497 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()