| /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 | depthwise_conv_grad_op.cc | 115 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; \
118 padding_, &out_rows, &pad_rows)); \
141 args.pad_rows = pad_rows; \
150 << ", pad_rows = " << pad_rows << ", pad_cols = " << pad_cols \
190 const int64 pad_rows = args.pad_rows; in CopyOutputBackpropRegion() local
197 static_cast<int64>(0), (in_r - filter_rows + pad_rows + stride) / stride); in CopyOutputBackpropRegion()
198 const int64 out_r_end = std::min(out_rows - 1, (in_r + pad_rows) / stride); in CopyOutputBackpropRegion()
217 const int64 f_r = in_r + pad_rows - out_r * stride; in CopyOutputBackpropRegion()
476 0, (in_r - args.filter_rows + args.pad_rows + stride) / stride); in DepthwiseConvBackpropInputReference()
478 std::min(args.out_rows - 1, (in_r + args.pad_rows) / stride); in DepthwiseConvBackpropInputReference()
[all …]
|
| D | pooling_ops_common.h | 68 int64 pad_rows; member
200 const int32 pad_rows = params.pad_rows; in SpatialMaxPool() local
223 const int32 hpad = h + pad_rows; in SpatialMaxPool()
277 params.pad_rows, params.pad_cols,
447 const int32 pad_rows = params.pad_rows;
470 const int32 hpad = h + pad_rows;
559 const int hpad = h + params.pad_rows;
|
| D | deep_conv2d.h | 75 int pad_rows; member
90 pad_rows(0), in Conv2DArgs()
|
| D | avgpooling_op.cc | 286 int64 out_height, out_width, pad_rows, pad_cols; in Compute() local
289 padding_, &out_height, &pad_rows)); in Compute()
300 col_stride, pad_rows, pad_cols](int64 start, int64 limit) { in Compute()
311 pad_rows, &rindex, &rsize)); in Compute()
526 int64 out_height, out_width, pad_rows, pad_cols; in Compute() local
529 padding_, &out_height, &pad_rows)); in Compute()
545 pad_rows, // pad_t in Compute()
|
| D | conv_ops_3d.cc | 224 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in launch() local
232 pad_rows = std::max<int64>( in launch()
296 const bool rows_odd = (pad_rows % 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()
379 .set_zero_padding(DimIndex::Y, pad_rows / 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
75 &pad_rows); in VerifyGet2dOutputSizeBoundaries()
86 int64 new_height, new_width, pad_rows, pad_cols; in VerifyGet2dOutputSizeValues() local
90 &pad_rows); in VerifyGet2dOutputSizeValues()
99 EXPECT_EQ(pad_struct.output.pad_top, pad_rows); in VerifyGet2dOutputSizeValues()
|
| D | depthwise_conv_op.cc | 348 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
351 padding_, &out_rows, &pad_rows)); in Compute()
383 << ", pad_rows = " << pad_rows << ", pad_cols = " << pad_cols in Compute()
421 args.pad_rows = pad_rows; in Compute()
|
| D | depthwise_conv_op.h | 35 int pad_rows; member
52 pad_rows(0), in DepthwiseArgs()
221 const int64 in_r_start = out_r * args.stride - args.pad_rows;
|
| D | extract_image_patches_op.cc | 87 int64 pad_rows = 0, pad_cols = 0; in Compute() local
90 padding_, &out_rows, &pad_rows)); in Compute()
|
| D | extract_volume_patches_op.cc | 110 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; in Compute() local
116 padding_, &out_rows, &pad_rows)); in Compute()
|
| D | depthwise_conv_op_gpu.h | 48 args.in_cols == args.out_cols && args.pad_rows >= 0 && in CanLaunchDepthwiseConv2dGPUSmall()
49 args.pad_rows < args.filter_rows && args.pad_cols >= 0 && in CanLaunchDepthwiseConv2dGPUSmall()
61 args.in_cols == args.out_cols && args.pad_rows >= 0 && in CanLaunchDepthwiseConv2dBackpropFilterGPUSmall()
62 args.pad_rows < args.filter_rows && args.pad_cols >= 0 && in CanLaunchDepthwiseConv2dBackpropFilterGPUSmall()
88 const int pad_height = args.pad_rows; in DepthwiseConv2dGPUKernelNHWC()
192 const int pad_height = args.pad_rows; in DepthwiseConv2dGPUKernelNHWCSmall()
328 const int pad_height = args.pad_rows; in DepthwiseConv2dGPUKernelNCHW()
476 const int pad_height = args.pad_rows; in DepthwiseConv2dGPUKernelNCHWSmall()
830 const int pad_height = args.pad_rows;
900 const int pad_height = args.pad_rows;
[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
191 args.pad_rows = pad_rows; 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
245 desc.pad_h = pad_rows; in Run()
|
| D | pooling_ops_common.cc | 84 padding, &out_height, &pad_rows)); in PoolParameters()
234 .set_vertical_padding(params.pad_rows) in Compute()
382 .set_vertical_padding(params.pad_rows) in Compute()
|
| D | pooling_ops_3d.h | 73 int64 pad_rows; member
|
| D | maxpooling_op.cc | 106 const int32 pad_rows = params.pad_rows; in SpatialMaxPoolWithArgMaxHelper() local
132 const int hpad = h + pad_rows; in SpatialMaxPoolWithArgMaxHelper()
341 params.window_cols, params.row_stride, params.col_stride, params.pad_rows, in MaxPoolingBackwardCustomKernel()
586 const int32 pad_rows = params.pad_rows; in SpatialMaxPoolGradGrad() local
609 int h_start = ph * row_stride - pad_rows; 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 | 89 padding, &out_height, &pad_rows)); in Pool3dParameters()
583 const int32 pad_rows = params.pad_rows; in launch() local
613 int h_start = ph * row_stride - pad_rows; 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
128 &out_rows, &pad_rows)); in BM_ConvFloat()
520 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in BM_ConvFloatDepthwise() local
522 &out_rows, &pad_rows)); in BM_ConvFloatDepthwise()
918 int64 out_height, out_width, pad_rows, pad_cols; in BM_AvgPoolBk() local
920 &out_height, &pad_rows)); in BM_AvgPoolBk()
1106 int64 out_height, out_width, pad_rows, pad_cols; in BM_MaxPoolBk() local
1108 &out_height, &pad_rows)); 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
523 padding_, &out_rows, &pad_rows)); in Compute()
|
| 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 | 825 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
828 padding_, &out_rows, &pad_rows)); 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 ()()
|
| D | mkl_conv_ops.cc | 525 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; in Compute() local
528 padding_, &out_rows, &pad_rows)); in Compute()
575 mkl_context.input_offset[1] = static_cast<int>(-pad_rows); in Compute()
|
| D | pooling_ops_3d_sycl.h | 83 pad_rows_(params.pad_rows), 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
214 &output_rows, &pad_rows)); in Compute()
|