Home
last modified time | relevance | path

Searched refs:pad_w (Results 1 – 12 of 12) 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()
Dxsmm_conv2d.cc125 d.v == w.d.v && d.pad_h == w.d.pad_h && d.pad_w == w.d.pad_w); in operator ==()
Dconv_grad_filter_ops.cc125 int col_stride, int pad_h, int pad_w, in operator ()()
138 int pad_w, TensorFormat data_format) const { in operator ()()
159 desc.pad_w = pad_w; in operator ()()
Dconv_grad_input_ops.cc131 int col_stride, int pad_h, int pad_w, in operator ()()
144 int col_stride, int pad_h, int pad_w, in operator ()()
165 desc.pad_w = pad_w; in operator ()()
Dconv_ops.cc246 desc.pad_w = pad_cols; in Run()
/external/libaom/libaom/test/
Dedge_detect_test.cc52 const int pad_w = w + 7; in pad_8tap_convolve() local
58 CONVERT_TO_BYTEPTR(aom_memalign(32, sizeof(uint16_t) * pad_w * pad_h)); in pad_8tap_convolve()
60 dst = (uint8_t *)aom_memalign(32, sizeof(uint8_t) * pad_w * pad_h); in pad_8tap_convolve()
63 for (int i = 0; i < pad_w; ++i) { in pad_8tap_convolve()
66 *CONVERT_TO_SHORTPTR(dst + i + j * pad_w) = v; in pad_8tap_convolve()
68 dst[i + j * pad_w] = v; in pad_8tap_convolve()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudnn_6_0.inc441 int pad_w, // zero-padding width
452 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
457 int* pad_w, // zero-padding width
468 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
1714 … int pad_w, // zero-padding width
1723 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
1728 int pad_w, // zero-padding width
1739 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
1745 int *pad_w, // zero-padding width
1754 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
[all …]
Dcudnn_7_0.inc484 int pad_w, /* zero-padding width */
495 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
500 int* pad_w, /* zero-padding width */
511 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
Dcudnn_7_3.inc483 int pad_w, /* zero-padding width */
493 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
499 int *pad_w, /* zero-padding width */
509 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
Dcudnn_7_1.inc484 int pad_w, /* zero-padding width */
495 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
500 int* pad_w, /* zero-padding width */
511 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
Dcudnn_7_4.inc483 int pad_w, /* zero-padding width */
493 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
499 int *pad_w, /* zero-padding width */
509 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc285 int pad_h = 0, pad_w = 0, u = 0, v = 0, dilation_h = 0, dilation_w = 0; in GetHashValue() local
286 wrap::miopenGetConvolutionDescriptor(conv_desc, &c_mode, &pad_h, &pad_w, &u, in GetHashValue()
293 tensorflow::Hash64Combine(hash_value, tensorflow::hash<int>()(pad_w)); in GetHashValue()