Searched defs:algorithm_config (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | convolution_thunk.cc | 75 se::dnn::AlgorithmConfig algorithm_config( in ExecuteOnStream() local
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 2146 const dnn::AlgorithmConfig& algorithm_config, bool is_profiling, in GetCudnnConvolutionForwardAlgorithm()
2310 const dnn::AlgorithmConfig& algorithm_config, in DoConvolveImpl()
2513 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolveImpl()
2932 const dnn::AlgorithmConfig& algorithm_config, in DoConvolve()
2960 const dnn::AlgorithmConfig& algorithm_config, in DoConvolve()
2979 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3001 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3024 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3046 const dnn::AlgorithmConfig& algorithm_config, in DoFusedConvolve()
3145 const dnn::AlgorithmConfig& algorithm_config, in DoConvolveBackwardDataImpl()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops_3d.cc | 660 AlgorithmConfig algorithm_config; in Compute() local
1027 AlgorithmConfig algorithm_config; in Compute() local
|
| D | conv_ops_3d.cc | 393 AlgorithmConfig algorithm_config; in launch() local
|
| D | matmul_op.cc | 296 AlgorithmConfig algorithm_config(kNoAlgorithm); in launch() local
|
| D | conv_grad_filter_ops.cc | 907 AlgorithmConfig algorithm_config; in operator ()() local
|
| D | conv_ops.cc | 707 AlgorithmConfig algorithm_config; in operator ()() local
|
| D | conv_grad_input_ops.cc | 956 AlgorithmConfig algorithm_config; in operator ()() local
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 541 dnn::AlgorithmConfig algorithm_config; in launch() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 580 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
619 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
657 const dnn::AlgorithmConfig &algorithm_config, in ThenFusedConvolveWithAlgorithm()
692 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
723 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveWithAlgorithm()
902 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveBackwardDataWithAlgorithm()
935 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveBackwardDataWithAlgorithm()
1038 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveBackwardFilterWithAlgorithm()
1099 const dnn::AlgorithmConfig &algorithm_config, in ThenConvolveBackwardFilterWithAlgorithm()
|