Home
last modified time | relevance | path

Searched refs:pad_h (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dxsmm_conv2d_test.cc39 int pad_h; member
123 int pad_h = param->pad_h; in naive_conv_fp() local
146 ij = oj * stride_h - pad_h; in naive_conv_fp()
202 int pad_h = pad; in TEST_F() local
205 int pad_h_in = pad_h; in TEST_F()
212 int ofh = (ifh + 2 * pad_h - kh) / stride_h + 1; in TEST_F()
255 naive_param.pad_h = pad_h; in TEST_F()
Dconv_grad_input_ops.h227 int col_stride, int pad_h, int pad_w,
240 int col_stride, int pad_h, int pad_w,
260 desc.pad_h = pad_h;
Dconv_grad_filter_ops.cc199 int col_stride, int pad_h, int pad_w, in operator ()()
211 int input_cols, int row_stride, int col_stride, int pad_h, in operator ()()
232 desc.pad_h = pad_h; in operator ()()
Dconv_ops.cc312 desc.pad_h = pad_rows; in Run()
/external/libaom/libaom/test/
Dedge_detect_test.cc55 const int pad_h = h + 7; in pad_8tap_convolve() local
60 CONVERT_TO_BYTEPTR(aom_memalign(32, sizeof(uint16_t) * pad_w * pad_h)); in pad_8tap_convolve()
62 dst = (uint8_t *)aom_memalign(32, sizeof(uint8_t) * pad_w * pad_h); in pad_8tap_convolve()
69 for (int j = 0; j < pad_h; ++j) { in pad_8tap_convolve()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc220 OpBuilder* builder, int32_t pad_h, in GetPadOpForConv2DFilter() argument
222 SmallVector<int32_t, 8> values = {pad_h, 0, pad_w, 0, 0, 0, 0, 0}; in GetPadOpForConv2DFilter()
227 std::vector<int64_t> pad_shape = {filter_shape[0] + pad_h, in GetPadOpForConv2DFilter()
282 int32_t pad_h = block_size - height % block_size; in HandleConv2DFilter() local
285 GetPadOpForConv2DFilter(filter_shape, filter, &builder, pad_h, pad_w); in HandleConv2DFilter()
288 height = height + pad_h; in HandleConv2DFilter()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudnn_6_0.inc447 int pad_h, // zero-padding height
459 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
465 int *pad_h, // zero-padding height
477 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1753 int pad_h, // zero-padding height
1766 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
1771 int pad_h, // zero-padding height
1784 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1790 int *pad_h, // zero-padding height
1803 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
[all …]
Dcudnn_7_0.inc491 cudnnConvolutionDescriptor_t convDesc, int pad_h, /* zero-padding height */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
509 int *pad_h, /* zero-padding height */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_3.inc491 cudnnConvolutionDescriptor_t convDesc, int pad_h, /* zero-padding height */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
509 int *pad_h, /* zero-padding height */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_1.inc491 cudnnConvolutionDescriptor_t convDesc, int pad_h, /* zero-padding height */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
509 int *pad_h, /* zero-padding height */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_4.inc491 cudnnConvolutionDescriptor_t convDesc, int pad_h, /* zero-padding height */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
509 int *pad_h, /* zero-padding height */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_8_0.inc1097 cudnnConvolutionDescriptor_t convDesc, int pad_h, /* zero-padding height */
1109 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1115 int *pad_h, /* zero-padding height */
1127 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_6.inc656 cudnnConvolutionDescriptor_t convDesc, int pad_h, /* zero-padding height */
668 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
674 int *pad_h, /* zero-padding height */
686 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,