Home
last modified time | relevance | path

Searched refs:explicit_paddings_attr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc1085 auto* explicit_paddings_attr = conv2d_bf_node->GetAttr("explicit_paddings"); in TEST_F() local
1086 ASSERT_NE(explicit_paddings_attr, nullptr); in TEST_F()
1087 ASSERT_EQ(explicit_paddings_attr->list().i_size(), 8); in TEST_F()
1088 EXPECT_EQ(explicit_paddings_attr->list().i(0), 0); in TEST_F()
1089 EXPECT_EQ(explicit_paddings_attr->list().i(1), 0); in TEST_F()
1090 EXPECT_EQ(explicit_paddings_attr->list().i(2), 0); in TEST_F()
1091 EXPECT_EQ(explicit_paddings_attr->list().i(3), 0); in TEST_F()
1092 EXPECT_EQ(explicit_paddings_attr->list().i(4), kPaddingTop); in TEST_F()
1093 EXPECT_EQ(explicit_paddings_attr->list().i(5), kPaddingBottom); in TEST_F()
1094 EXPECT_EQ(explicit_paddings_attr->list().i(6), kPaddingLeft); in TEST_F()
[all …]
Dgeneric_layout_optimizer_transposer.cc718 const auto* explicit_paddings_attr = node->GetAttr(kAttrExplicitPaddings); in UpdateNode() local
719 if (explicit_paddings_attr != nullptr && explicit_paddings_attr->has_list() && in UpdateNode()
720 explicit_paddings_attr->list().i_size() > 0) { in UpdateNode()
721 AttrValue explicit_paddings_attr_copy(*explicit_paddings_attr); in UpdateNode()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc4142 ArrayRef<Attribute> explicit_paddings_attr = in matchAndRewrite() local
4144 explicit_paddings.reserve(explicit_paddings_attr.size()); in matchAndRewrite()
4145 for (Attribute explicit_padding : explicit_paddings_attr) in matchAndRewrite()
4325 ArrayRef<Attribute> explicit_paddings_attr = in matchAndRewrite() local
4327 explicit_paddings.reserve(explicit_paddings_attr.size()); in matchAndRewrite()
4328 for (Attribute explicit_padding : explicit_paddings_attr) in matchAndRewrite()