HomeSort by relevance Sort by last modified time
    Searched refs:pad_bottom (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
ops_util_test.cc 46 int pad_bottom; member in struct:tensorflow::__anon45438::OpsUtilTest::padding_struct::__anon45440
105 int64 new_height, new_width, pad_top, pad_bottom, pad_left, pad_right; local
109 &pad_top, &pad_bottom);
119 EXPECT_EQ(pad_struct.output.pad_bottom, pad_bottom);
mkl_conv_ops.h 391 int64 pad_top = 0, pad_bottom = 0, pad_left, pad_right; local
400 pad_bottom = static_cast<int64>((*pad_r)[0]);
408 padding_type, &out_rows, &pad_top, &pad_bottom));
419 padding_, &out_rows, &pad_top, &pad_bottom));
432 *pad_r = {static_cast<int>(pad_bottom), static_cast<int>(pad_right)};
438 *pad_r = {static_cast<int>(pad_D2), static_cast<int>(pad_bottom),
conv_grad_input_ops.cc 346 int64 pad_top, pad_bottom; variable
353 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
361 if (pad_left == pad_right && pad_top == pad_bottom) {
375 int64 pad_top, pad_bottom; variable
383 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
490 dims.spatial_dims[1].filter_size, pad_top, pad_left, pad_bottom,
503 auto shard = [&context, &dims, &pad_top, &pad_left, &pad_bottom,
522 pad_bottom, pad_right, dims.spatial_dims[0].stride,
    [all...]
mkl_pooling_ops_common.h 392 int64 pad_bottom; member in struct:tensorflow::MklPoolParameters
419 pad_bottom(0),
533 *padding_right = memory::dims({static_cast<int>(pool_params->pad_bottom),
548 static_cast<int>(pool_params->pad_bottom),
conv_grad_filter_ops.cc 265 int64 pad_top, pad_bottom; variable
272 &dims.spatial_dims[0].output_size, &pad_top, &pad_bottom));
281 if (pad_left == pad_right && pad_top == pad_bottom) {
369 &pad_bottom, &pad_right, &input_offset,
380 dims.spatial_dims[1].filter_size, pad_top, pad_left, pad_bottom,
    [all...]
mkl_pooling_ops_common.cc 353 padding, &out_height, &pad_top, &pad_bottom));
427 mkl_params->in_offset[3] = -params.pad_bottom;
mkl_conv_ops.cc 1194 int64 pad_bottom, pad_right; local
    [all...]
  /external/tensorflow/tensorflow/python/kernel_tests/
atrous_conv2d_test.py 130 pad_bottom = 0 if height % rate == 0 else rate - height % rate
132 pad = [[0, pad_bottom], [0, pad_right]]
  /external/tensorflow/tensorflow/python/ops/
nn_ops.py     [all...]
  /external/tensorflow/tensorflow/go/op/
wrappers.go 786 // paddings = [[pad_top, pad_bottom], [pad_left, pad_right]]
790 // height_pad = pad_top + height + pad_bottom
    [all...]

Completed in 253 milliseconds