Searched refs:transpose_node (Results 1 – 2 of 2) sorted by relevance
1136 const NodeDef* transpose_node = nullptr; in TEST_F() local1139 EXPECT_EQ(transpose_node, nullptr); in TEST_F()1141 transpose_node = &node; in TEST_F()1144 EXPECT_NE(transpose_node, nullptr); in TEST_F()1148 EXPECT_EQ(NodeName(node.input(0)), transpose_node->name()); in TEST_F()1276 const NodeDef* transpose_node = nullptr; in TEST_F() local1280 EXPECT_EQ(transpose_node, nullptr); in TEST_F()1282 transpose_node = &node; in TEST_F()1293 EXPECT_NE(transpose_node, nullptr); in TEST_F()1295 EXPECT_EQ(NodeName(transpose_node->input(0)), reverse_node->name()); in TEST_F()[all …]
2251 bool IsInnerMatrixTransposeNode(const NodeDef& transpose_node, in IsInnerMatrixTransposeNode() argument2253 if (transpose_node.op() != "Transpose" && in IsInnerMatrixTransposeNode()2254 transpose_node.op() != "ConjugateTranspose") { in IsInnerMatrixTransposeNode()2257 const NodeDef* perm_node = node_map->GetNode(transpose_node.input(1)); in IsInnerMatrixTransposeNode()