Searched refs:padding_list_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_ops_test.cc | 1463 int padding_list_[4]; member in tensorflow::MklPadWithFusedConv2DOpTest
1524 {0, 0, padding_list_[0], padding_list_[1], in RunMklPadWithFusedConv2DAndBias()
1525 padding_list_[2], padding_list_[3], 0, 0}, in RunMklPadWithFusedConv2DAndBias()
1552 {0, 0, padding_list_[0], padding_list_[1], in RunMklPadWithFusedConv2DAndBiasRelu()
1553 padding_list_[2], padding_list_[3], 0, 0}, in RunMklPadWithFusedConv2DAndBiasRelu()
1582 &padding, {0, 0, padding_list_[0], padding_list_[1], padding_list_[2], in RunMklFusedConv2DWithPadOp()
1583 padding_list_[3], 0, 0}); in RunMklFusedConv2DWithPadOp()
1648 padding_list_[0] = top; in SetPaddingList()
1649 padding_list_[1] = bottom; in SetPaddingList()
1650 padding_list_[2] = left; in SetPaddingList()
[all …]
|
| D | mkl_conv_ops.cc | 433 OP_REQUIRES_OK(context, context->GetAttr("padding_list", &padding_list_)); in MklConvOp()
437 context->GetAttr("explicit_paddings", &padding_list_)); in MklConvOp()
521 for (auto const& padding_val : padding_list_) { in Compute()
755 paddings = padding_list_.data(); in PadWithConvFusion()
940 std::vector<Tpadding> padding_list_; member in tensorflow::MklConvOp
|