| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | neon_depthwise_conv_op.cc | 90 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
96 padding_, &out_cols, &pad_cols)); in Compute()
114 << ", pad_cols = " << pad_cols << ", output: [" << batch << ", " in Compute()
137 bias_neon_dims, stride, pad_cols, pad_rows, depth_multiplier, in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_grad_op.cc | 115 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; \
121 padding_, &out_cols, &pad_cols)); \
142 args.pad_cols = pad_cols; \
150 << ", pad_rows = " << pad_rows << ", pad_cols = " << pad_cols \
191 const int64 pad_cols = args.pad_cols; in CopyOutputBackpropRegion() local
200 static_cast<int64>(0), (in_c - filter_cols + pad_cols + stride) / stride); in CopyOutputBackpropRegion()
201 const int64 out_c_end = std::min(out_cols - 1, (in_c + pad_cols) / stride); in CopyOutputBackpropRegion()
219 const int64 f_c = in_c + pad_cols - out_c * stride; in CopyOutputBackpropRegion()
483 (in_c - args.filter_cols + args.pad_cols + stride) / stride); in DepthwiseConvBackpropInputReference()
485 std::min(args.out_cols - 1, (in_c + args.pad_cols) / stride); in DepthwiseConvBackpropInputReference()
[all …]
|
| D | pooling_ops_common.h | 69 int64 pad_cols; member
201 const int32 pad_cols = params.pad_cols; in SpatialMaxPool() local
224 const int32 wpad = w + pad_cols; in SpatialMaxPool()
277 params.pad_rows, params.pad_cols,
448 const int32 pad_cols = params.pad_cols;
471 const int32 wpad = w + pad_cols;
560 const int wpad = w + params.pad_cols;
|
| D | deep_conv2d.h | 76 int pad_cols; member
91 pad_cols(0), in Conv2DArgs()
|
| D | avgpooling_op.cc | 286 int64 out_height, out_width, pad_rows, pad_cols; in Compute() local
292 padding_, &out_width, &pad_cols)); in Compute()
300 col_stride, pad_rows, pad_cols](int64 start, int64 limit) { in Compute()
321 pad_cols, &cindex, &csize)); in Compute()
526 int64 out_height, out_width, pad_rows, pad_cols; in Compute() local
532 padding_, &out_width, &pad_cols)); in Compute()
546 pad_cols, // pad_l in Compute()
|
| D | conv_ops_3d.cc | 224 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in launch() local
234 pad_cols = std::max<int64>( in launch()
297 const bool cols_odd = (pad_cols % 2 != 0); in launch()
348 CHECK(pad_rows >= 0 && pad_cols >= 0 && pad_planes >= 0) in launch()
349 << "Negative paddings: (" << pad_rows << ", " << pad_cols << ", " in launch()
378 .set_zero_padding(DimIndex::X, pad_cols / 2) in launch()
426 {{pad_planes, pad_rows, pad_cols}}, in launch()
|
| D | ops_util_test.cc | 71 int64 new_height, new_width, pad_rows, pad_cols; in VerifyGet2dOutputSizeBoundaries() local
80 &pad_cols); in VerifyGet2dOutputSizeBoundaries()
86 int64 new_height, new_width, pad_rows, pad_cols; in VerifyGet2dOutputSizeValues() local
95 &pad_cols); in VerifyGet2dOutputSizeValues()
100 EXPECT_EQ(pad_struct.output.pad_left, pad_cols); in VerifyGet2dOutputSizeValues()
|
| D | depthwise_conv_op.cc | 348 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
354 padding_, &out_cols, &pad_cols)); in Compute()
383 << ", pad_rows = " << pad_rows << ", pad_cols = " << pad_cols in Compute()
422 args.pad_cols = pad_cols; in Compute()
|
| D | depthwise_conv_op.h | 36 int pad_cols; member
53 pad_cols(0), in DepthwiseArgs()
222 const int64 in_c_start = out_c * args.stride - args.pad_cols;
|
| D | extract_image_patches_op.cc | 87 int64 pad_rows = 0, pad_cols = 0; in Compute() local
93 padding_, &out_cols, &pad_cols)); in Compute()
|
| D | extract_volume_patches_op.cc | 110 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in Compute() local
119 padding_, &out_cols, &pad_cols)); in Compute()
|
| D | depthwise_conv_op_gpu.h | 49 args.pad_rows < args.filter_rows && args.pad_cols >= 0 && in CanLaunchDepthwiseConv2dGPUSmall()
50 args.pad_cols < args.filter_cols && in CanLaunchDepthwiseConv2dGPUSmall()
62 args.pad_rows < args.filter_rows && args.pad_cols >= 0 && in CanLaunchDepthwiseConv2dBackpropFilterGPUSmall()
63 args.pad_cols < args.filter_cols && block_height <= args.in_rows && in CanLaunchDepthwiseConv2dBackpropFilterGPUSmall()
89 const int pad_width = args.pad_cols; in DepthwiseConv2dGPUKernelNHWC()
193 const int pad_width = args.pad_cols; in DepthwiseConv2dGPUKernelNHWCSmall()
329 const int pad_width = args.pad_cols; in DepthwiseConv2dGPUKernelNCHW()
477 const int pad_width = args.pad_cols; in DepthwiseConv2dGPUKernelNCHWSmall()
831 const int pad_width = args.pad_cols;
901 const int pad_width = args.pad_cols;
[all …]
|
| D | conv_ops.cc | 157 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run() argument
173 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run() argument
192 args.pad_cols = pad_cols; in Run()
214 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run() argument
228 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run() argument
246 desc.pad_w = pad_cols; in Run()
|
| D | pooling_ops_common.cc | 87 padding, &out_width, &pad_cols)); in PoolParameters()
235 .set_horizontal_padding(params.pad_cols) in Compute()
383 .set_horizontal_padding(params.pad_cols) in Compute()
|
| D | pooling_ops_3d.h | 72 int64 pad_cols; member
|
| D | maxpooling_op.cc | 107 const int32 pad_cols = params.pad_cols; in SpatialMaxPoolWithArgMaxHelper() local
133 const int wpad = w + pad_cols; in SpatialMaxPoolWithArgMaxHelper()
342 params.pad_cols, out_backprop.flat<T>().data(), output->flat<T>().data(), in MaxPoolingBackwardCustomKernel()
587 const int32 pad_cols = params.pad_cols; in SpatialMaxPoolGradGrad() local
611 int w_start = pw * col_stride - pad_cols; in SpatialMaxPoolGradGrad()
734 params.col_stride, params.pad_rows, params.pad_cols, in Compute()
1307 params.row_stride, params.col_stride, params.pad_rows, params.pad_cols, in launch()
1326 params.row_stride, params.col_stride, params.pad_rows, params.pad_cols, in launch()
|
| D | pooling_ops_3d.cc | 92 padding, &out_width, &pad_cols)); in Pool3dParameters()
584 const int32 pad_cols = params.pad_cols; in launch() local
615 int w_start = pw * col_stride - pad_cols; in launch()
815 params.pad_planes, params.pad_rows, params.pad_cols, in launch()
|
| D | nn_ops_test.cc | 126 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloat() local
130 &out_cols, &pad_cols)); in BM_ConvFloat()
520 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloatDepthwise() local
524 &out_cols, &pad_cols)); in BM_ConvFloatDepthwise()
918 int64 out_height, out_width, pad_rows, pad_cols; in BM_AvgPoolBk() local
922 &out_width, &pad_cols)); in BM_AvgPoolBk()
1106 int64 out_height, out_width, pad_rows, pad_cols; in BM_MaxPoolBk() local
1110 &out_width, &pad_cols)); in BM_MaxPoolBk()
|
| D | conv_ops_using_gemm.cc | 520 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
526 padding_, &out_cols, &pad_cols)); in Compute()
|
| D | quantized_conv_ops.cc | 540 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
546 padding_, &out_cols, &pad_cols)); in Compute()
|
| D | conv_ops_fused_image_transform.cc | 825 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
831 padding_, &out_cols, &pad_cols)); in Compute()
|
| D | deep_conv2d.cc | 797 args.pad_cols - sc * tile_stride_cols; in operator ()()
1111 const int64 col_pad = args.pad_cols; in operator ()()
|
| D | mkl_conv_ops.cc | 525 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
531 padding_, &out_cols, &pad_cols)); in Compute()
574 mkl_context.input_offset[0] = static_cast<int>(-pad_cols); in Compute()
|
| D | pooling_ops_3d_sycl.h | 84 pad_cols_(params.pad_cols) {} in SYCL3DPoolParams()
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 211 int64 output_rows = 0, output_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
217 &output_cols, &pad_cols)); in Compute()
|