| /external/tensorflow/tensorflow/core/kernels/ |
| D | dilation_ops.cc | 67 int* rate_cols, int64* pad_top, int64* pad_left, in ParseSizes() argument
108 padding, out_rows, pad_top)); in ParseSizes()
128 int64 pad_top = 0, pad_left = 0; in Compute() local
131 &rate_rows, &rate_cols, &pad_top, &pad_left, &out_rows, in Compute()
152 pad_top, pad_left, output->tensor<T, 4>()); in Compute()
166 int stride_cols, int rate_rows, int rate_cols, int pad_top, in operator ()()
183 int h_beg = h_out * stride_rows - pad_top; in operator ()()
227 int64 pad_top = 0, pad_left = 0; in Compute() local
230 &rate_rows, &rate_cols, &pad_top, &pad_left, &out_rows, in Compute()
258 stride_cols, rate_rows, rate_cols, pad_top, pad_left, in Compute()
[all …]
|
| D | dilation_ops_gpu.cu.cc | 43 int rate_cols, int pad_top, int pad_left, T* __restrict__ output_ptr) { in DilationKernel() argument
52 int h_beg = h_out * stride_rows - pad_top; in DilationKernel()
82 int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, in DilationBackpropInputKernel() argument
92 int h_beg = h_out * stride_rows - pad_top; in DilationBackpropInputKernel()
132 int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, in DilationBackpropFilterKernel() argument
142 int h_beg = h_out * stride_rows - pad_top; in DilationBackpropFilterKernel()
183 int stride_cols, int rate_rows, int rate_cols, int pad_top, in operator ()()
204 rate_cols, pad_top, pad_left, output.data())); in operator ()()
214 int rate_cols, int pad_top, int pad_left, in operator ()()
245 stride_rows, stride_cols, rate_rows, rate_cols, pad_top, pad_left, in operator ()()
[all …]
|
| D | pooling_ops_common.cc | 87 Status CheckPaddingSize(int64 window_rows, int64 window_cols, int64 pad_top, in CheckPaddingSize() argument
89 if (!FastBoundsCheck(pad_top, window_rows)) { in CheckPaddingSize()
90 return errors::InvalidArgument("Top padding ", pad_top, in CheckPaddingSize()
155 GetExplicitPaddingForDim(explicit_paddings, data_format, 'H', &pad_top, in PoolParameters()
159 OP_REQUIRES_OK(context, CheckPaddingSize(window_rows, window_cols, pad_top, in PoolParameters()
166 padding, &out_height, &pad_top, &pad_bottom)); in PoolParameters()
292 int64 vertical_padding = params.pad_top; in Compute()
295 if (padding == EXPLICIT && (params.pad_top != params.pad_bottom || in Compute()
301 std::min(params.pad_top, params.pad_bottom); in Compute()
307 std::abs(params.pad_top - params.pad_bottom); in Compute()
[all …]
|
| D | dilation_ops.h | 31 int stride_cols, int rate_rows, int rate_cols, int pad_top,
45 int rate_cols, int pad_top, int pad_left,
59 int rate_cols, int pad_top, int pad_left,
|
| D | ops_util_test.cc | 47 int pad_top; member
101 EXPECT_EQ(pad_struct.output.pad_top, pad_rows); in VerifyGet2dOutputSizeValues()
107 int64 new_height, new_width, pad_top, pad_bottom, pad_left, pad_right; in VerifyGet2dOutputVerboseSizeValues() local
111 &pad_top, &pad_bottom); in VerifyGet2dOutputVerboseSizeValues()
120 EXPECT_EQ(pad_struct.output.pad_top, pad_top); in VerifyGet2dOutputVerboseSizeValues()
|
| D | conv_grad_input_ops.h | 558 int64 pad_top, pad_bottom;
565 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
573 if (pad_left == pad_right && pad_top == pad_bottom) {
581 static_cast<int>(pad_top), static_cast<int>(pad_left),
587 int64 pad_top, pad_bottom;
591 pad_top = explicit_paddings_[2];
601 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
708 dims.spatial_dims[1].filter_size, pad_top, pad_left, pad_bottom,
721 auto shard = [&context, &dims, &pad_top, &pad_left, &pad_bottom,
739 dims.spatial_dims[1].filter_size, pad_top, pad_left,
|
| D | pooling_ops_common.h | 76 int64 pad_top; member
238 const int32 pad_top = params.pad_top; in SpatialMaxPool() local
261 const int32 hpad = h + pad_top; in SpatialMaxPool()
317 params.pad_top, params.pad_left,
499 const int32 pad_top = params.pad_top;
522 const int32 hpad = h + pad_top;
611 const int hpad = h + params.pad_top;
|
| D | depthwise_conv_op.cc | 374 int64 out_rows = 0, out_cols = 0, pad_top = 0, pad_bottom = 0, pad_left = 0, in Compute() local
377 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'H', &pad_top, in Compute()
384 &out_rows, &pad_top, &pad_bottom)); in Compute()
422 << ", pad_top = " << pad_top << ", pad_left = " << pad_left in Compute()
460 args.pad_rows = pad_top; in Compute()
|
| D | conv_2d.h | 146 Eigen::DenseIndex pad_top) {
157 .slice(Eigen::DSizes<Eigen::DenseIndex, 4>{0, pad_left, pad_top, 0},
174 Eigen::DenseIndex pad_top) {
180 .slice(Eigen::DSizes<Eigen::DenseIndex, 4>{0, pad_left, pad_top, 0},
|
| D | maxpooling_op.cc | 114 const int32 pad_top = params.pad_top; in SpatialMaxPoolWithArgMaxHelper() local
140 const int hpad = h + pad_top; in SpatialMaxPoolWithArgMaxHelper()
420 int64 pad_top, pad_bottom, pad_left, pad_right; in Compute() local
423 /*pad_top=*/&pad_top, in Compute()
594 const int32 pad_top = params.pad_top; in SpatialMaxPoolGradGrad() local
617 int h_start = ph * row_stride - pad_top; in SpatialMaxPoolGradGrad()
747 params.col_stride, params.pad_top, params.pad_left, in Compute()
1363 params.row_stride, params.col_stride, params.pad_top, params.pad_left, in launch()
1382 params.row_stride, params.col_stride, params.pad_top, params.pad_left, in launch()
|
| D | conv_grad_filter_ops.cc | 458 int64 pad_top, pad_bottom; in Compute() local
461 pad_top = explicit_paddings_[2]; in Compute()
471 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom)); in Compute()
480 if (pad_left == pad_right && pad_top == pad_bottom) { in Compute()
487 static_cast<int>(pad_top), static_cast<int>(pad_left), in Compute()
567 auto shard = [&input_data, &col_buffer_data, &dims, &pad_top, &pad_left, in Compute()
579 dims.spatial_dims[1].filter_size, pad_top, pad_left, pad_bottom, in Compute()
|
| D | depthwise_conv_grad_op.cc | 120 int64 out_rows = 0, out_cols = 0, pad_top = 0, pad_bottom = 0, pad_left = 0, \
123 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'H', &pad_top, \
130 &out_rows, &pad_top, &pad_bottom)); \
153 args.pad_rows = pad_top; \
162 << ", pad_rows = " << pad_top << ", pad_cols = " << pad_left \
653 << ", pad_rows = " << pad_top << ", pad_cols = " << pad_left in Compute()
1150 << ", pad_rows = " << pad_top << ", pad_cols = " << pad_left in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 397 int64 pad_top = 0, pad_bottom = 0, pad_left = 0, pad_right = 0; variable
404 pad_top = static_cast<int64>((*pad_l)[0]);
414 padding_type, &out_rows, &pad_top, &pad_bottom));
427 padding_, &out_rows, &pad_top, &pad_bottom));
440 *pad_l = {static_cast<int>(pad_top), static_cast<int>(pad_left)};
445 *pad_l = {static_cast<int>(pad_D1), static_cast<int>(pad_top),
|
| D | mkl_pooling_ops_common.h | 384 int64 pad_top; member
411 pad_top(0), in MklPoolParameters()
521 *padding_left = memory::dims({static_cast<int>(pool_params->pad_top), in PoolParamsToDims()
535 static_cast<int>(pool_params->pad_top), in PoolParamsToDims()
|
| D | mkl_pooling_ops_common.cc | 338 padding, &out_height, &pad_top, &pad_bottom)); in Init()
|
| D | mkl_conv_ops.cc | 776 int64 pad_top = 0, pad_left = 0; in PadWithConvFusion() local
780 pad_top = paddings[2]; in PadWithConvFusion()
785 pad_top = paddings[4]; in PadWithConvFusion()
792 padding_left = {static_cast<int>(pad_top), static_cast<int>(pad_left)}; in PadWithConvFusion()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SpaceToBatch.pbtxt | 15 paddings = [[pad_top, pad_bottom], [pad_left, pad_right]]
19 height_pad = pad_top + height + pad_bottom
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 2728 pad_top = pad_height // 2
2729 pad_bottom = pad_height - pad_top
2733 pad_top = 0
2741 in_height = output_shape[1] + pad_top + pad_bottom
2769 batch_to_space_crop = [[pad_top, pad_bottom + pad_bottom_extra],
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 13868 paddings = [[pad_top, pad_bottom], [pad_left, pad_right]]
13872 height_pad = pad_top + height + pad_bottom
|