Home
last modified time | relevance | path

Searched refs:pad_h (Results 1 – 12 of 12) 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()
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 ()()
137 int input_cols, int row_stride, int col_stride, int pad_h, in operator ()()
158 desc.pad_h = pad_h; 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 ()()
164 desc.pad_h = pad_h; in operator ()()
Dconv_ops.cc245 desc.pad_h = pad_rows; in Run()
/external/libaom/libaom/test/
Dedge_detect_test.cc53 const int pad_h = h + 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()
62 for (int j = 0; j < pad_h; ++j) { in pad_8tap_convolve()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcudnn_6_0.inc440 int pad_h, // zero-padding height
452 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
456 int* pad_h, // zero-padding height
468 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
1713 … int pad_h, // zero-padding height
1723 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
1727 int pad_h, // zero-padding height
1739 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
1744 int *pad_h, // zero-padding height
1754 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode);
[all …]
Dcudnn_7_0.inc483 int pad_h, /* zero-padding height */
495 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
499 int* pad_h, /* zero-padding height */
511 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
Dcudnn_7_3.inc482 int pad_h, /* zero-padding height */
493 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
498 int *pad_h, /* zero-padding height */
509 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
Dcudnn_7_1.inc483 int pad_h, /* zero-padding height */
495 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
499 int* pad_h, /* zero-padding height */
511 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
Dcudnn_7_4.inc482 int pad_h, /* zero-padding height */
493 return func_ptr(convDesc, pad_h, pad_w, u, v, dilation_h, dilation_w, mode, computeType);
498 int *pad_h, /* zero-padding height */
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()
291 tensorflow::Hash64Combine(hash_value, tensorflow::hash<int>()(pad_h)); in GetHashValue()