Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dxsmm_conv2d_test.cc40 int pad_w; member
124 int pad_w = param->pad_w; in naive_conv_fp() local
148 ii = oi * stride_w - pad_w; in naive_conv_fp()
203 int pad_w = pad; in TEST_F() local
206 int pad_w_in = pad_w; in TEST_F()
213 int ofw = (ifw + 2 * pad_w - kw) / stride_w + 1; in TEST_F()
256 naive_param.pad_w = pad_w; 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,
261 desc.pad_w = pad_w;
Dconv_grad_filter_ops.cc199 int col_stride, int pad_h, int pad_w, in operator ()()
212 int pad_w, TensorFormat data_format) const { in operator ()()
233 desc.pad_w = pad_w; in operator ()()
Dconv_ops.cc313 desc.pad_w = pad_cols; in Run()
/external/libaom/libaom/test/
Dedge_detect_test.cc54 const int pad_w = w + 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()
70 for (int i = 0; i < pad_w; ++i) { in pad_8tap_convolve()
73 *CONVERT_TO_SHORTPTR(dst + i + j * pad_w) = v; in pad_8tap_convolve()
75 dst[i + j * pad_w] = static_cast<uint8_t>(v); in pad_8tap_convolve()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc221 int32_t pad_w) { in GetPadOpForConv2DFilter() argument
222 SmallVector<int32_t, 8> values = {pad_h, 0, pad_w, 0, 0, 0, 0, 0}; in GetPadOpForConv2DFilter()
228 filter_shape[1] + pad_w, filter_shape[2], in GetPadOpForConv2DFilter()
283 int32_t pad_w = block_size - width % block_size; in HandleConv2DFilter() local
285 GetPadOpForConv2DFilter(filter_shape, filter, &builder, pad_h, pad_w); in HandleConv2DFilter()
289 width = width + pad_w; in HandleConv2DFilter()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudnn_6_0.inc448 int pad_w, // zero-padding width
459 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
466 int *pad_w, // zero-padding width
477 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1754 int pad_w, // zero-padding width
1766 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
1772 int pad_w, // zero-padding width
1784 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1791 int *pad_w, // zero-padding width
1803 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
[all …]
Dcudnn_7_0.inc492 int pad_w, /* zero-padding width */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
510 int *pad_w, /* zero-padding width */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_3.inc492 int pad_w, /* zero-padding width */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
510 int *pad_w, /* zero-padding width */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_1.inc492 int pad_w, /* zero-padding width */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
510 int *pad_w, /* zero-padding width */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_4.inc492 int pad_w, /* zero-padding width */
503 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
510 int *pad_w, /* zero-padding width */
521 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_8_0.inc1098 int pad_w, /* zero-padding width */
1109 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
1116 int *pad_w, /* zero-padding width */
1127 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
Dcudnn_7_6.inc657 int pad_w, /* zero-padding width */
668 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,
675 int *pad_w, /* zero-padding width */
686 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode,