| /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
93 padding_, &out_rows, &pad_rows)); in Compute()
113 << "]; stride = " << stride << ", pad_rows = " << pad_rows in Compute()
137 bias_neon_dims, stride, pad_cols, pad_rows, depth_multiplier, in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deep_conv2d.h | 75 int pad_rows; member
90 pad_rows(0), in Conv2DArgs()
|
| D | avgpooling_op.cc | 323 int64 out_height, out_width, pad_rows, pad_cols; in Compute() local
326 padding_, &out_height, &pad_rows)); in Compute()
337 col_stride, pad_rows, pad_cols](int64 start, int64 limit) { in Compute()
348 pad_rows, &rindex, &rsize)); in Compute()
583 int64 out_height, out_width, pad_rows, pad_cols; in Compute() local
586 padding_, &out_height, &pad_rows)); in Compute()
602 pad_rows, // pad_t in Compute()
|
| D | ops_util_test.cc | 73 int64 new_height, new_width, pad_rows, pad_cols; in VerifyGet2dOutputSizeBoundaries() local
77 &pad_rows); in VerifyGet2dOutputSizeBoundaries()
88 int64 new_height, new_width, pad_rows, pad_cols; in VerifyGet2dOutputSizeValues() local
92 &pad_rows); in VerifyGet2dOutputSizeValues()
101 EXPECT_EQ(pad_struct.output.pad_top, pad_rows); in VerifyGet2dOutputSizeValues()
|
| D | conv_ops_3d.cc | 234 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in launch() local
242 pad_rows = std::max<int64>( in launch()
308 const bool rows_odd = (pad_rows % 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()
426 .set_zero_padding(DimIndex::Y, pad_rows / 2) in launch()
490 {{pad_planes, pad_rows, pad_cols}}, in launch()
|
| D | depthwise_conv_op.h | 35 int pad_rows; // Amount of padding to the top of the input member
52 pad_rows(0), in DepthwiseArgs()
213 const int64 in_r_start = out_r * args.stride - args.pad_rows;
|
| D | depthwise_conv_grad_op.cc | 153 args.pad_rows = pad_top; \
202 const int64 pad_rows = args.pad_rows; in CopyOutputBackpropRegion() local
209 static_cast<int64>(0), (in_r - filter_rows + pad_rows + stride) / stride); in CopyOutputBackpropRegion()
210 const int64 out_r_end = std::min(out_rows - 1, (in_r + pad_rows) / stride); in CopyOutputBackpropRegion()
229 const int64 f_r = in_r + pad_rows - out_r * stride; in CopyOutputBackpropRegion()
488 0, (in_r - args.filter_rows + args.pad_rows + stride) / stride); in DepthwiseConvBackpropInputReference()
490 std::min(args.out_rows - 1, (in_r + args.pad_rows) / stride); in DepthwiseConvBackpropInputReference()
500 int f_r = in_r + args.pad_rows - out_r * stride; in DepthwiseConvBackpropInputReference()
997 const int in_r_start = out_r * args.stride - args.pad_rows; in DepthwiseConvBackpropFilterReference()
|
| D | depthwise_conv_op_gpu.h | 58 args.in_cols == args.out_cols && args.pad_rows >= 0 &&
59 args.pad_rows < args.filter_rows && args.pad_cols >= 0 &&
71 args.in_cols == args.out_cols && args.pad_rows >= 0 &&
72 args.pad_rows < args.filter_rows && args.pad_cols >= 0 &&
101 const int pad_height = args.pad_rows;
209 const int pad_height = args.pad_rows;
348 const int pad_height = args.pad_rows;
500 const int pad_height = args.pad_rows;
821 const int pad_height = args.pad_rows;
891 const int pad_height = args.pad_rows;
[all …]
|
| D | pooling_ops_3d.h | 73 int64 pad_rows; 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
258 args.pad_rows = pad_rows; 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
312 desc.pad_h = pad_rows; in Run()
|
| D | pooling_ops_3d.cc | 84 padding, &out_height, &pad_rows)); in Pool3dParameters()
579 const int32 pad_rows = params.pad_rows; in launch() local
609 int h_start = ph * row_stride - pad_rows; 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
525 padding_, &out_rows, &pad_rows)); 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
130 &out_rows, &pad_rows)); in BM_ConvFloat()
537 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloatDepthwise() local
539 &out_rows, &pad_rows)); in BM_ConvFloatDepthwise()
941 int64 out_height, out_width, pad_rows, pad_cols; in BM_AvgPoolBk() local
943 &out_height, &pad_rows)); in BM_AvgPoolBk()
1129 int64 out_height, out_width, pad_rows, pad_cols; in BM_MaxPoolBk() local
1131 &out_height, &pad_rows)); 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
543 padding_, &out_rows, &pad_rows)); 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
825 padding_, &out_rows, &pad_rows)); in Compute()
|
| D | depthwise_conv_op.cc | 460 args.pad_rows = pad_top; in Compute()
|
| D | deep_conv2d.cc | 792 in_r + args.pad_rows - sr * tile_stride_rows; in operator ()()
1110 const int64 row_pad = args.pad_rows; 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
119 padding_, &out_rows, &pad_rows)); in Compute()
|
| D | extract_image_patches_op.cc | 90 int64 pad_rows = 0, pad_cols = 0; in Compute() local
93 padding_, &out_rows, &pad_rows)); in Compute()
|