Searched defs:padding_rows (Results 1 - 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dconv_grad_ops_3d.cc530 int padding_rows = 0, padding_cols = 0, padding_planes = 0; variable
537 padding_rows = std::max<int>(
540 const bool rows_odd = (padding_rows % 2 != 0);
559 CHECK(padding_rows >= 0 && padding_cols >= 0 && padding_planes >= 0)
560 << "Negative paddings: (" << padding_rows << ", " << padding_cols
587 .set_zero_padding(DimIndex::Y, padding_rows / 2)
652 {{padding_planes, padding_rows, padding_cols}},
884 int padding_rows = 0, padding_cols = 0, padding_planes = 0; variable
891 padding_rows = std::max<int>(
894 bool rows_odd = (padding_rows
[all...]
H A Dconv_grad_filter_ops.cc657 const int padding_rows = local
677 bool rows_odd = (padding_rows % 2 != 0);
784 CHECK(padding_rows >= 0 && padding_cols >= 0)
785 << "Negative row or col paddings: (" << padding_rows << ", "
809 .set_zero_padding_height(padding_rows / 2)
902 {{padding_rows, // padding_rows
H A Dconv_grad_input_ops.cc728 const int padding_rows = local
748 bool rows_odd = (padding_rows % 2 != 0);
838 CHECK(padding_rows >= 0 && padding_cols >= 0)
839 << "Negative row or col paddings: (" << padding_rows << ", "
863 .set_zero_padding_height(padding_rows / 2)
951 {{padding_rows, // padding_rows
H A Dconv_ops.cc558 int padding_rows = 0; local
579 padding_rows =
585 const bool rows_odd = (padding_rows % 2 != 0);
628 CHECK(padding_rows >= 0 && padding_cols >= 0)
629 << "Negative row or col paddings: (" << padding_rows << ", "
653 .set_zero_padding_height(padding_rows / 2)
702 {{padding_rows, // padding_rows
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
H A Dfused_conv2d_bias_activation_op.cc346 int padding_rows = 0; local
360 padding_rows = std::max<int>(
368 AdjustPaddingForCudnn(padding_rows, is_int8x4, filter_rows, &padding_rows,
475 CHECK_EQ(0, padding_rows % 2);
479 .set_zero_padding_height(padding_rows / 2)
534 {{padding_rows, padding_cols}},

Completed in 334 milliseconds