Searched refs:padding_list_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_fused_ops_test.cc | 484 int padding_list_[4]; member in tensorflow::MklPadWithFusedConv2DOpTest
545 {0, 0, padding_list_[0], padding_list_[1], in RunMklPadWithFusedConv2DAndBias()
546 padding_list_[2], padding_list_[3], 0, 0}, in RunMklPadWithFusedConv2DAndBias()
573 {0, 0, padding_list_[0], padding_list_[1], in RunMklPadWithFusedConv2DAndBiasRelu()
574 padding_list_[2], padding_list_[3], 0, 0}, in RunMklPadWithFusedConv2DAndBiasRelu()
603 &padding, {0, 0, padding_list_[0], padding_list_[1], padding_list_[2], in RunMklFusedConv2DWithPadOp()
604 padding_list_[3], 0, 0}); in RunMklFusedConv2DWithPadOp()
647 padding_list_[0] = top; in SetPaddingList()
648 padding_list_[1] = bottom; in SetPaddingList()
649 padding_list_[2] = left; in SetPaddingList()
[all …]
|
| D | mkl_conv_ops.cc | 864 OP_REQUIRES_OK(context, context->GetAttr("padding_list", &padding_list_)); in MklConvOp()
947 for (auto const& padding_val : padding_list_) { in Compute()
1173 paddings = padding_list_.data(); in PadWithConvFusion()
1284 std::vector<Tpadding> padding_list_; member in tensorflow::MklConvOp
|