Searched refs:pad_r (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.h | 316 memory::dims* pad_l, memory::dims* pad_r, bool pad_enabled = false,
321 CHECK_NOTNULL(pad_r);
400 pad_bottom = static_cast<int64>((*pad_r)[0]);
401 pad_right = static_cast<int64>((*pad_r)[1]);
432 *pad_r = {static_cast<int>(pad_bottom), static_cast<int>(pad_right)};
438 *pad_r = {static_cast<int>(pad_D2), static_cast<int>(pad_bottom),
480 memory::dims* pad_r, bool is_depthwise) { in GetOutputAndPadSizeInMklOrder() argument
484 CHECK_NOTNULL(pad_r); in GetOutputAndPadSizeInMklOrder()
503 output_dims_mkl_order, pad_l, pad_r, in GetOutputAndPadSizeInMklOrder()
520 memory::dims* pad_l, memory::dims* pad_r, bool pad_enabled = false,
[all …]
|
| D | eigen_pooling_test.cc | 401 const int pad_r = output_rows - input_rows + patch_rows - 1; in TEST() local
407 const int dr = pad_r / 2; in TEST()
479 const int pad_r = output_rows - input_rows + patch_rows - 1; in TEST() local
485 const int dr = pad_r / 2; in TEST()
|
| D | conv_grad_ops_3d.cc | 69 const int pad_r, const int stride_p, const int stride_h, in Col2im() argument
73 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im()
118 const int pad_r, const int stride_p, const int stride_h, in Im2col() argument
122 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col()
|
| D | eigen_spatial_convolutions_test.cc | 1172 const int pad_r = (out_height - 1) * stride - in_rows + kern_height; in TEST() local
1178 const int dr = pad_r / 2; in TEST()
1243 const int pad_r = (out_height - 1) * stride - in_rows + kern_height; in TEST() local
1249 const int dr = pad_r / 2; in TEST()
|
| D | conv_grad_filter_ops.cc | 67 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Im2col() argument
70 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col()
|
| D | conv_grad_input_ops.cc | 67 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Col2im() argument
70 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im()
|