Home
last modified time | relevance | path

Searched refs:pad_attr (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dmerge_padding_with.cc58 PadAttributes pad_attr = in ApplyToNodesSequence() local
61 if (pad_attr.type != PaddingContentType::ZEROS) { in ApplyToNodesSequence()
64 if (pad_attr.appended.c != 0 || pad_attr.prepended.c != 0 || in ApplyToNodesSequence()
65 pad_attr.appended.b != 0 || pad_attr.prepended.b != 0) { in ApplyToNodesSequence()
78 node_attr->padding.appended.h += pad_attr.appended.h; in ApplyToNodesSequence()
79 node_attr->padding.appended.w += pad_attr.appended.w; in ApplyToNodesSequence()
80 node_attr->padding.prepended.h += pad_attr.prepended.h; in ApplyToNodesSequence()
81 node_attr->padding.prepended.w += pad_attr.prepended.w; in ApplyToNodesSequence()
84 absl::StrCat("Added padding: prepended = {h = ", pad_attr.prepended.h, in ApplyToNodesSequence()
85 ", w = ", pad_attr.prepended.w, "}, appended = { h = ", in ApplyToNodesSequence()
[all …]
Dmake_padding.cc67 PadAttributes pad_attr; in ApplyToNode() local
68 pad_attr.type = PaddingContentType::ZEROS; in ApplyToNode()
69 pad_attr.appended = BHWC(0, 0, 0, 0); in ApplyToNode()
70 pad_attr.prepended = BHWC(0, 0, 0, 0); in ApplyToNode()
71 BHWC* p = first ? &pad_attr.prepended : &pad_attr.appended; in ApplyToNode()
92 node->operation.attributes = pad_attr; in ApplyToNode()
Dmerge_padding_with_test.cc128 PadAttributes pad_attr; in TEST() local
129 pad_attr.prepended = BHWC(0, 0, 0, 0); in TEST()
130 pad_attr.appended = BHWC(0, 0, 0, 32); in TEST()
131 pad_node->operation.attributes = pad_attr; in TEST()
166 PadAttributes pad_attr; in TEST() local
167 pad_attr.prepended = BHWC(0, 0, 0, 0); in TEST()
168 pad_attr.appended = BHWC(0, 0, 0, 32); in TEST()
169 pad_node->operation.attributes = pad_attr; in TEST()
Dmake_padding_test.cc74 auto pad_attr = absl::any_cast<PadAttributes>(pad_node->operation.attributes); in TEST() local
75 EXPECT_EQ(BHWC(0, 0, 0, 0), pad_attr.prepended); in TEST()
76 EXPECT_EQ(BHWC(0, 5, 0, 0), pad_attr.appended); in TEST()