Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1010 const dnn::AlgorithmConfig& algorithm_config, in CudnnRnnDescriptor()
1037 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in Create()
1117 const dnn::AlgorithmConfig& algorithm_config() const { in algorithm_config() function in stream_executor::gpu::CudnnRnnDescriptor
1820 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in createRnnDescriptor()
2343 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionForwardAlgorithm()
2392 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardDataAlgorithm()
2441 const dnn::AlgorithmConfig& algorithm_config, in GetCudnnConvolutionBackwardFilterAlgorithm()
2632 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
2934 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolveImpl()
3354 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_3d.cc435 AlgorithmConfig algorithm_config; in launch() local
Dmatmul_op.cc291 AlgorithmConfig algorithm_config(kNoAlgorithm); in launch() local
Dconv_grad_ops_3d.cc1357 AlgorithmConfig algorithm_config; in Compute() local
1763 AlgorithmConfig algorithm_config; in Compute() local
Dconv_grad_filter_ops.cc839 AlgorithmConfig algorithm_config; in operator ()() local
Dconv_grad_input_ops.cc951 AlgorithmConfig algorithm_config; in operator ()() local
Dconv_ops.cc847 AlgorithmConfig algorithm_config; in operator ()() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc451 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
489 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
528 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
566 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
601 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
643 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
685 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
830 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveBackwardDataWithAlgorithm()
875 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveBackwardDataWithAlgorithm()
920 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveBackwardDataWithAlgorithm()
[all …]
Dstream_executor_pimpl.cc385 const dnn::AlgorithmConfig &algorithm_config, float dropout, uint64 seed, in createRnnDescriptor()
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.cc692 dnn::AlgorithmConfig algorithm_config; in launch() local
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc2279 const dnn::AlgorithmConfig& algorithm_config, float dropout, uint64 seed, in createRnnDescriptor()
2598 const dnn::AlgorithmConfig& algorithm_config, in DoPrepareForConvolution()
3156 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3173 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3191 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3208 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()