| /external/tensorflow/tensorflow/core/kernels/ |
| deep_conv2d.h | 76 int pad_cols; member in struct:tensorflow::Conv2DArgs
91 pad_cols(0),
|
| pooling_ops_3d.h | 72 int64 pad_cols; member in struct:tensorflow::Pool3dParameters
|
| extract_image_patches_op.cc | 87 int64 pad_rows = 0, pad_cols = 0; variable
93 padding_, &out_cols, &pad_cols));
|
| depthwise_conv_op.h | 36 int pad_cols; member in struct:tensorflow::DepthwiseArgs
53 pad_cols(0),
222 const int64 in_c_start = out_c * args.stride - args.pad_cols;
|
| depthwise_conv_op.cc | 341 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; variable
347 padding_, &out_cols, &pad_cols));
365 << ", pad_cols = " << pad_cols << ", output: [" << batch << ", "
394 args.pad_cols = pad_cols;
|
| ops_util_test.cc | 71 int64 new_height, new_width, pad_rows, pad_cols; local
80 &pad_cols);
86 int64 new_height, new_width, pad_rows, pad_cols; local
95 &pad_cols);
100 EXPECT_EQ(pad_struct.output.pad_left, pad_cols);
|
| conv_ops.cc | 138 int filter_cols, int pad_rows, int pad_cols, int out_rows,
154 int filter_cols, int pad_rows, int pad_cols, int out_rows,
173 args.pad_cols = pad_cols;
195 int filter_cols, int pad_rows, int pad_cols, int out_rows,
209 int filter_cols, int pad_rows, int pad_cols, int out_rows,
227 desc.pad_w = pad_cols;
373 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; variable
379 stride_cols, padding_, &out_cols, &pad_cols));
407 filter_rows, filter_cols, pad_rows, pad_cols, out_rows, out_cols
[all...] |
| avgpooling_op.cc | 279 int64 out_height, out_width, pad_rows, pad_cols; variable
285 padding_, &out_width, &pad_cols));
293 col_stride, pad_rows, pad_cols](int64 start, int64 limit) {
314 pad_cols, &cindex, &csize));
513 int64 out_height, out_width, pad_rows, pad_cols; variable
519 padding_, &out_width, &pad_cols));
533 pad_cols, // pad_l
|
| conv_ops_3d.cc | 186 int64 pad_planes = 0, pad_rows = 0, pad_cols = 0; local
196 pad_cols = std::max<int64>(
258 const bool cols_odd = (pad_cols % 2 != 0);
309 CHECK(pad_rows >= 0 && pad_cols >= 0 && pad_planes >= 0)
310 << "Negative paddings: (" << pad_rows << ", " << pad_cols << ", "
336 .set_zero_padding(DimIndex::X, pad_cols / 2)
384 {{pad_planes, pad_rows, pad_cols}},
|
| depthwise_conv_grad_op.cc | 112 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; \
118 padding_, &out_cols, &pad_cols)); \
139 args.pad_cols = pad_cols; \
147 << ", pad_rows = " << pad_rows << ", pad_cols = " << pad_cols \
188 const int64 pad_cols = args.pad_cols; local
197 static_cast<int64>(0), (in_c - filter_cols + pad_cols + stride) / stride);
198 const int64 out_c_end = std::min(out_cols - 1, (in_c + pad_cols) / stride)
[all...] |
| pooling_ops_common.h | 69 int64 pad_cols; member in struct:tensorflow::PoolParameters
201 const int32 pad_cols = params.pad_cols;
224 const int32 wpad = w + pad_cols;
277 params.pad_rows, params.pad_cols,
448 const int32 pad_cols = params.pad_cols;
471 const int32 wpad = w + pad_cols;
560 const int wpad = w + params.pad_cols;
|
| pooling_ops_common.cc | 72 padding, &out_width, &pad_cols));
194 .set_horizontal_padding(params.pad_cols)
337 .set_horizontal_padding(params.pad_cols)
|
| maxpooling_op.cc | 96 const int32 pad_cols = params.pad_cols;
122 const int wpad = w + pad_cols;
328 params.pad_cols, out_backprop.flat<T>().data(), output->flat<T>().data(),
573 const int32 pad_cols = params.pad_cols;
597 int w_start = pw * col_stride - pad_cols;
720 params.col_stride, params.pad_rows, params.pad_cols,
[all...] |
| depthwise_conv_op_gpu.cu.cc | 46 args.pad_rows < args.filter_rows && args.pad_cols >= 0 &&
47 args.pad_cols < args.filter_cols &&
59 args.pad_rows < args.filter_rows && args.pad_cols >= 0 &&
60 args.pad_cols < args.filter_cols && block_height <= args.in_rows &&
86 const int pad_width = args.pad_cols;
187 const int pad_width = args.pad_cols;
322 const int pad_width = args.pad_cols;
467 const int pad_width = args.pad_cols;
[all...] |
| conv_ops_using_gemm.cc | 520 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; variable
526 padding_, &out_cols, &pad_cols));
|
| pooling_ops_3d.cc | 92 padding, &out_width, &pad_cols));
584 const int32 pad_cols = params.pad_cols;
615 int w_start = pw * col_stride - pad_cols;
[all...] |
| nn_ops_test.cc | 126 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; local
130 &out_cols, &pad_cols));
520 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; local
524 &out_cols, &pad_cols));
918 int64 out_height, out_width, pad_rows, pad_cols; local
1106 int64 out_height, out_width, pad_rows, pad_cols; local
[all...] |
| quantized_conv_ops.cc | 540 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; variable
546 padding_, &out_cols, &pad_cols));
|
| mkl_conv_ops.cc | 174 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; variable
180 padding_, &out_cols, &pad_cols));
223 mkl_context.input_offset[0] = static_cast<int>(-pad_cols);
[all...] |
| conv_ops_fused.cc | 824 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; variable
[all...] |
| deep_conv2d.cc | 796 args.pad_cols - sc * tile_stride_cols;
[all...] |
| pooling_ops_3d_sycl.h | 84 pad_cols_(params.pad_cols) {}
[all...] |
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| neon_depthwise_conv_op.cc | 90 int64 out_rows = 0, out_cols = 0, pad_rows = 0, pad_cols = 0; variable
96 padding_, &out_cols, &pad_cols));
114 << ", pad_cols = " << pad_cols << ", output: [" << batch << ", "
137 bias_neon_dims, stride, pad_cols, pad_rows, depth_multiplier,
|
| /external/tensorflow/tensorflow/python/ops/ |
| array_grad.py | 755 pad_cols = ((cols_out - 1) * stride_h + ksize_c_eff - cols_in) // 2
761 pad_cols = (cols_out - 1) * stride_h + ksize_c_eff - cols_in
763 pad_rows, pad_cols = max(0, pad_rows), max(0, pad_cols)
777 r_low, c_low = row_steps[i] - pad_rows, col_steps[j] - pad_cols
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| fused_conv2d_bias_activation_op.cc | 201 int64 output_rows = 0, output_cols = 0, pad_rows = 0, pad_cols = 0; variable
207 &output_cols, &pad_cols));
|
