| /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 | deep_conv2d.h | 76 int pad_cols; member
91 pad_cols(0), in Conv2DArgs()
|
| D | avgpooling_op.cc | 323 int64 out_height, out_width, pad_rows, pad_cols; in Compute() local
329 padding_, &out_width, &pad_cols)); in Compute()
337 col_stride, pad_rows, pad_cols](int64 start, int64 limit) { in Compute()
358 pad_cols, &cindex, &csize)); in Compute()
583 int64 out_height, out_width, pad_rows, pad_cols; in Compute() local
589 padding_, &out_width, &pad_cols)); in Compute()
603 pad_cols, // pad_l in Compute()
|
| D | ops_util_test.cc | 73 int64 new_height, new_width, pad_rows, pad_cols; in VerifyGet2dOutputSizeBoundaries() local
82 &pad_cols); in VerifyGet2dOutputSizeBoundaries()
88 int64 new_height, new_width, pad_rows, pad_cols; in VerifyGet2dOutputSizeValues() local
97 &pad_cols); in VerifyGet2dOutputSizeValues()
102 EXPECT_EQ(pad_struct.output.pad_left, pad_cols); in VerifyGet2dOutputSizeValues()
|
| D | conv_ops_3d.cc | 234 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in launch() local
244 pad_cols = std::max<int64>( in launch()
309 const bool cols_odd = (pad_cols % 2 != 0); in launch()
394 CHECK(pad_rows >= 0 && pad_cols >= 0 && pad_planes >= 0) in launch()
395 << "Negative paddings: (" << pad_rows << ", " << pad_cols << ", " in launch()
425 .set_zero_padding(DimIndex::X, pad_cols / 2) in launch()
490 {{pad_planes, pad_rows, pad_cols}}, in launch()
|
| D | depthwise_conv_op.h | 36 int pad_cols; // Amount of padding to the left of the input member
53 pad_cols(0), in DepthwiseArgs()
214 const int64 in_c_start = out_c * args.stride - args.pad_cols;
|
| D | depthwise_conv_grad_op.cc | 154 args.pad_cols = pad_left; \
203 const int64 pad_cols = args.pad_cols; in CopyOutputBackpropRegion() local
212 static_cast<int64>(0), (in_c - filter_cols + pad_cols + stride) / stride); in CopyOutputBackpropRegion()
213 const int64 out_c_end = std::min(out_cols - 1, (in_c + pad_cols) / stride); in CopyOutputBackpropRegion()
231 const int64 f_c = in_c + pad_cols - out_c * stride; in CopyOutputBackpropRegion()
495 (in_c - args.filter_cols + args.pad_cols + stride) / stride); in DepthwiseConvBackpropInputReference()
497 std::min(args.out_cols - 1, (in_c + args.pad_cols) / stride); in DepthwiseConvBackpropInputReference()
501 int f_c = in_c + args.pad_cols - out_c * stride; in DepthwiseConvBackpropInputReference()
998 const int in_c_start = out_c * args.stride - args.pad_cols; in DepthwiseConvBackpropFilterReference()
|
| D | depthwise_conv_op_gpu.h | 59 args.pad_rows < args.filter_rows && args.pad_cols >= 0 &&
60 args.pad_cols < args.filter_cols &&
72 args.pad_rows < args.filter_rows && args.pad_cols >= 0 &&
73 args.pad_cols < args.filter_cols && block_height <= args.in_rows &&
102 const int pad_width = args.pad_cols;
210 const int pad_width = args.pad_cols;
349 const int pad_width = args.pad_cols;
501 const int pad_width = args.pad_cols;
822 const int pad_width = args.pad_cols;
892 const int pad_width = args.pad_cols;
[all …]
|
| D | pooling_ops_3d.h | 72 int64 pad_cols; member
|
| D | conv_ops.cc | 224 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run() argument
240 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run() argument
259 args.pad_cols = pad_cols; in Run()
281 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run() argument
295 int filter_cols, int pad_rows, int pad_cols, int out_rows, in Run() argument
313 desc.pad_w = pad_cols; in Run()
|
| D | pooling_ops_3d.cc | 87 padding, &out_width, &pad_cols)); in Pool3dParameters()
580 const int32 pad_cols = params.pad_cols; in launch() local
611 int w_start = pw * col_stride - pad_cols; in launch()
811 params.pad_planes, params.pad_rows, params.pad_cols, in launch()
|
| D | conv_ops_using_gemm.cc | 522 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
528 padding_, &out_cols, &pad_cols)); in Compute()
|
| D | nn_ops_test.cc | 128 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloat() local
132 &out_cols, &pad_cols)); in BM_ConvFloat()
537 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloatDepthwise() local
541 &out_cols, &pad_cols)); in BM_ConvFloatDepthwise()
941 int64 out_height, out_width, pad_rows, pad_cols; in BM_AvgPoolBk() local
945 &out_width, &pad_cols)); in BM_AvgPoolBk()
1129 int64 out_height, out_width, pad_rows, pad_cols; in BM_MaxPoolBk() local
1133 &out_width, &pad_cols)); in BM_MaxPoolBk()
|
| 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 | 822 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
828 padding_, &out_cols, &pad_cols)); in Compute()
|
| D | depthwise_conv_op.cc | 461 args.pad_cols = pad_left; 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 ()()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_volume_patches_op.cc | 113 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in Compute() local
122 padding_, &out_cols, &pad_cols)); in Compute()
|
| D | extract_image_patches_op.cc | 90 int64 pad_rows = 0, pad_cols = 0; in Compute() local
96 padding_, &out_cols, &pad_cols)); in Compute()
|