| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 371 se::dnn::ConvolutionDescriptor conv_desc(3); in launch() local
372 conv_desc.set_dilation_rate(DimIndex::X, dilations[2]) in launch()
459 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch()
488 filter_ptr, conv_desc, output_desc, in launch()
|
| D | conv_grad_ops_3d.cc | 1274 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute() local
1275 conv_desc.set_dilation_rate(DimIndex::X, dims.dilation(2)) in Compute()
1377 conv_desc, input_desc, &in_backprop_ptr, &scratch_allocator, in Compute()
1413 conv_desc, input_desc, &in_backprop_ptr, &scratch_allocator, in Compute()
1673 se::dnn::ConvolutionDescriptor conv_desc(3); in Compute() local
1674 conv_desc.set_dilation_rate(DimIndex::X, dims.dilation(2)) in Compute()
1783 conv_desc, filter_desc, &filter_backprop_ptr, in Compute()
1819 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in Compute()
|
| D | conv_ops.cc | 780 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
781 conv_desc.set_vertical_dilation_rate(row_dilation) in operator ()()
869 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()()
898 filter_ptr, conv_desc, output_desc, in operator ()()
|
| D | conv_grad_filter_ops.cc | 733 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
734 conv_desc.set_vertical_dilation_rate(dims.spatial_dims[0].dilation) in operator ()()
857 conv_desc, filter_desc, &filter_backprop_ptr, in operator ()()
887 input_desc, input_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|
| D | conv_grad_input_ops.cc | 849 se::dnn::ConvolutionDescriptor conv_desc; in operator ()() local
850 conv_desc.set_vertical_dilation_rate(dims.spatial_dims[0].dilation) in operator ()()
969 conv_desc, input_desc, &in_backprop_ptr, &scratch_allocator, in operator ()()
997 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|
| D | conv_ops_fused_impl.h | 695 se::dnn::ConvolutionDescriptor conv_desc;
696 conv_desc.set_vertical_dilation_rate(dimensions.dilation_rows)
784 conv_desc, // conv
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 611 dnn::ConvolutionDescriptor conv_desc; in launch() local
614 conv_desc.set_vertical_filter_stride(row_stride) in launch()
724 filter_desc, filter_ptr, conv_desc, side_input_ptr, in launch()
758 filter_ptr, conv_desc, side_input_ptr, side_input_scale, in launch()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 283 uint64 GetHashValue(miopenConvolutionDescriptor_t conv_desc) { in GetHashValue() argument
286 wrap::miopenGetConvolutionDescriptor(conv_desc, &c_mode, &pad_h, &pad_w, &u, in GetHashValue()
|