| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 474 dnn::ConvolutionDescriptor conv_desc; local
477 conv_desc.set_vertical_filter_stride(row_stride)
559 filter_desc, filter_ptr, conv_desc, side_input_ptr,
598 filter_ptr, conv_desc, side_input_ptr, side_input_scale,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| conv_ops.cc | 648 perftools::gputools::dnn::ConvolutionDescriptor conv_desc; local
649 conv_desc.set_vertical_dilation_rate(row_dilation)
723 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc,
760 filter_ptr, conv_desc, output_desc,
[all...] |
| conv_grad_filter_ops.cc | 804 perftools::gputools::dnn::ConvolutionDescriptor conv_desc; local
[all...] |
| conv_grad_input_ops.cc | 858 perftools::gputools::dnn::ConvolutionDescriptor conv_desc; local
[all...] |
| mkl_conv_ops.cc | 607 auto conv_desc = convolution_forward::desc( variable
613 convolution_forward::primitive_desc(conv_desc, cpu_engine);
628 auto conv_desc = convolution_forward::desc( variable
634 convolution_forward::primitive_desc(conv_desc, cpu_engine);
[all...] |
