Searched refs:padding_cols (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 482 int padding_cols = 0; in launch() local
497 padding_cols = std::max<int>( in launch()
505 AdjustPaddingForCudnn(padding_cols, is_int8x4, filter_cols, &padding_cols, in launch()
613 CHECK_EQ(0, padding_cols % 2); in launch()
617 .set_zero_padding_width(padding_cols / 2); in launch()
673 {{padding_rows, padding_cols}}, in launch()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops_3d.cc | 1230 int padding_cols = dims.SpatialPadding(padding_, 2); in Compute() local
1233 const bool cols_odd = (padding_cols % 2 != 0); in Compute()
1251 CHECK(padding_rows >= 0 && padding_cols >= 0 && padding_planes >= 0) in Compute()
1252 << "Negative paddings: (" << padding_rows << ", " << padding_cols in Compute()
1281 .set_zero_padding(DimIndex::X, padding_cols / 2) in Compute()
1349 {{padding_planes, padding_rows, padding_cols}}, in Compute()
1622 int padding_cols = dims.SpatialPadding(padding_, 2); in Compute() local
1625 const bool cols_odd = (padding_cols % 2 != 0); in Compute()
1647 CHECK(padding_rows >= 0 && padding_cols >= 0 && padding_planes >= 0) in Compute()
1648 << "Negative paddings: (" << padding_rows << ", " << padding_cols in Compute()
[all …]
|
| D | conv_ops_fused_impl.h | 591 int64 padding_cols = 0;
618 padding_cols = std::max<int>(
622 const bool cols_odd = (padding_cols % 2 != 0);
667 CHECK(padding_cols >= 0) << "Negative padding cols"; // Crash OK
701 .set_zero_padding_width(padding_cols / 2)
767 padding_cols}}, // padding_cols
|