| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_3d.cc | 365 se::dnn::FilterDescriptor filter_desc(3); in launch() local
366 filter_desc.set_spatial_dim(DimIndex::X, filter_cols) in launch()
459 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in launch()
487 ->ThenConvolveWithAlgorithm(input_desc, input_ptr, filter_desc, in launch()
|
| D | conv_grad_ops_3d.cc | 1268 se::dnn::FilterDescriptor filter_desc(3); in Compute() local
1269 filter_desc.set_spatial_dim(DimIndex::X, dims.filter_size(2)) in Compute()
1376 filter_desc, filter_ptr, output_desc, out_backprop_ptr, in Compute()
1412 filter_desc, filter_ptr, output_desc, out_backprop_ptr, in Compute()
1667 se::dnn::FilterDescriptor filter_desc(3); in Compute() local
1668 filter_desc.set_spatial_dim(DimIndex::X, dims.filter_size(2)) in Compute()
1783 conv_desc, filter_desc, &filter_backprop_ptr, in Compute()
1820 filter_desc, &filter_backprop_ptr, &scratch_allocator, in Compute()
|
| D | conv_ops.cc | 775 se::dnn::FilterDescriptor filter_desc; in operator ()() local
776 filter_desc.set_input_filter_height(patch_rows) in operator ()()
869 input_desc, input_ptr, filter_desc, filter_ptr, conv_desc, in operator ()()
897 ->ThenConvolveWithAlgorithm(input_desc, input_ptr, filter_desc, in operator ()()
|
| D | conv_grad_filter_ops.cc | 728 se::dnn::FilterDescriptor filter_desc; in operator ()() local
729 filter_desc.set_input_filter_height(dims.spatial_dims[0].filter_size) in operator ()()
857 conv_desc, filter_desc, &filter_backprop_ptr, in operator ()()
888 filter_desc, &filter_backprop_ptr, &scratch_allocator, in operator ()()
|
| D | conv_grad_input_ops.cc | 844 se::dnn::FilterDescriptor filter_desc; in operator ()() local
845 filter_desc.set_input_filter_height(dims.spatial_dims[0].filter_size) in operator ()()
968 filter_desc, filter_ptr, output_desc, out_backprop_ptr, in operator ()()
997 filter_desc, filter_ptr, output_desc, out_backprop_ptr, conv_desc, in operator ()()
|
| D | conv_ops_fused_impl.h | 684 se::dnn::FilterDescriptor filter_desc;
685 filter_desc.set_input_filter_height(patch_rows)
783 filter_desc, filter_ptr, // filter
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 587 dnn::FilterDescriptor filter_desc; in launch() local
588 filter_desc.set_input_filter_height(filter_rows) in launch()
724 filter_desc, filter_ptr, conv_desc, side_input_ptr, in launch()
757 conv_input_desc, conv_input_ptr, conv_input_scale, filter_desc, in launch()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1177 FilterDescriptor filter_desc = CreateFilterDescriptor(); in Create() local
1180 /*filterDesc=*/filter_desc.get(), /*dataType=*/data_type, in Create()
1216 /*wDesc=*/filter_desc.get(), in Create()
1223 /*wDesc=*/filter_desc.get(), in Create()
1245 return CudnnRnnParamsDescriptor(std::move(filter_desc), params_size_in_bytes, in Create()
|