Searched refs:pad_r (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 322 memory::dims* pad_l, memory::dims* pad_r, bool pad_enabled = false,
327 DCHECK(pad_r);
406 pad_bottom = static_cast<int64>((*pad_r)[0]);
407 pad_right = static_cast<int64>((*pad_r)[1]);
441 *pad_r = {static_cast<int>(pad_bottom), static_cast<int>(pad_right)};
447 *pad_r = {static_cast<int>(pad_D2), static_cast<int>(pad_bottom),
489 memory::dims* pad_r, bool is_depthwise) { in GetOutputAndPadSizeInMklOrder() argument
493 DCHECK(pad_r); in GetOutputAndPadSizeInMklOrder()
512 output_dims_mkl_order, pad_l, pad_r, in GetOutputAndPadSizeInMklOrder()
529 memory::dims* pad_l, memory::dims* pad_r, bool pad_enabled = false,
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| 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_input_ops.h | 79 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Col2im() argument
82 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im()
|
| 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_ops_3d.cc | 78 const int pad_r, const int stride_p, const int stride_h, in Col2im() argument
82 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im()
127 const int pad_r, const int stride_p, const int stride_h, in Im2col() argument
131 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col()
|
| D | conv_grad_filter_ops.cc | 72 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Im2col() argument
75 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 5664 const int pad_t, const int pad_l, const int pad_b, const int pad_r, in Col2im() argument
5668 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im()
|