Searched refs:padding_rows (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| D | fused_conv2d_bias_activation_op.cc | 481 int padding_rows = 0; in launch() local
495 padding_rows = std::max<int>( in launch()
503 AdjustPaddingForCudnn(padding_rows, is_int8x4, filter_rows, &padding_rows, in launch()
612 CHECK_EQ(0, padding_rows % 2); in launch()
616 .set_zero_padding_height(padding_rows / 2) in launch()
673 {{padding_rows, padding_cols}}, in launch()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_ops_3d.cc | 1229 int padding_rows = dims.SpatialPadding(padding_, 1); in Compute() local
1232 const bool rows_odd = (padding_rows % 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()
1282 .set_zero_padding(DimIndex::Y, padding_rows / 2) in Compute()
1349 {{padding_planes, padding_rows, padding_cols}}, in Compute()
1621 int padding_rows = dims.SpatialPadding(padding_, 1); in Compute() local
1624 const bool rows_odd = (padding_rows % 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 | 590 int64 padding_rows = 0;
615 padding_rows = std::max<int>(
621 const bool rows_odd = (padding_rows % 2 != 0);
666 CHECK(padding_rows >= 0) << "Negative padding rows"; // Crash OK
700 .set_zero_padding_height(padding_rows / 2)
766 {{padding_rows, // padding_rows
|