| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_common.cc | 88 int64 pad_bottom, int64 pad_left, int64 pad_right) { in CheckPaddingSize() argument
95 if (!FastBoundsCheck(pad_bottom, window_rows)) { in CheckPaddingSize()
96 return errors::InvalidArgument("Bottom padding ", pad_bottom, in CheckPaddingSize()
156 &pad_bottom); in PoolParameters()
160 pad_bottom, pad_left, pad_right)); in PoolParameters()
166 padding, &out_height, &pad_top, &pad_bottom)); in PoolParameters()
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()
321 const int64 input_pad_bottom = params.pad_bottom - common_padding_rows; in Compute()
[all …]
|
| 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) {
587 int64 pad_top, pad_bottom;
592 pad_bottom = explicit_paddings_[3];
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,
740 pad_bottom, pad_right, dims.spatial_dims[0].stride,
|
| D | ops_util_test.cc | 48 int pad_bottom; member
107 int64 new_height, new_width, pad_top, pad_bottom, pad_left, pad_right; in VerifyGet2dOutputVerboseSizeValues() local
111 &pad_top, &pad_bottom); in VerifyGet2dOutputVerboseSizeValues()
121 EXPECT_EQ(pad_struct.output.pad_bottom, pad_bottom); in VerifyGet2dOutputVerboseSizeValues()
|
| 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
378 &pad_bottom); in Compute()
384 &out_rows, &pad_top, &pad_bottom)); in Compute()
|
| D | conv_grad_filter_ops.cc | 458 int64 pad_top, pad_bottom; in Compute() local
462 pad_bottom = explicit_paddings_[3]; 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()
568 &pad_bottom, &pad_right, &input_offset, 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, \
124 &pad_bottom); \
130 &out_rows, &pad_top, &pad_bottom)); \
|
| D | pooling_ops_common.h | 77 int64 pad_bottom; member
|
| D | maxpooling_op.cc | 420 int64 pad_top, pad_bottom, pad_left, pad_right; in Compute() local
424 /*pad_bottom=*/&pad_bottom); 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
406 pad_bottom = static_cast<int64>((*pad_r)[0]);
414 padding_type, &out_rows, &pad_top, &pad_bottom));
427 padding_, &out_rows, &pad_top, &pad_bottom));
441 *pad_r = {static_cast<int>(pad_bottom), static_cast<int>(pad_right)};
447 *pad_r = {static_cast<int>(pad_D2), static_cast<int>(pad_bottom),
|
| D | mkl_pooling_ops_common.h | 385 int64 pad_bottom; member
412 pad_bottom(0), in MklPoolParameters()
523 *padding_right = memory::dims({static_cast<int>(pool_params->pad_bottom), in PoolParamsToDims()
538 static_cast<int>(pool_params->pad_bottom), in PoolParamsToDims()
|
| D | mkl_pooling_ops_common.cc | 338 padding, &out_height, &pad_top, &pad_bottom)); in Init()
|
| D | mkl_conv_ops.cc | 777 int64 pad_bottom = 0, pad_right = 0; in PadWithConvFusion() local
781 pad_bottom = paddings[3]; in PadWithConvFusion()
786 pad_bottom = paddings[5]; in PadWithConvFusion()
793 padding_right = {static_cast<int>(pad_bottom), static_cast<int>(pad_right)}; 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/kernel_tests/ |
| D | atrous_conv2d_test.py | 129 pad_bottom = 0 if height % rate == 0 else rate - height % rate
131 pad = [[0, pad_bottom], [0, pad_right]]
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 2729 pad_bottom = pad_height - pad_top
2734 pad_bottom = 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
|