Searched defs:transpose_node (Results 1 – 4 of 4) sorted by relevance
1571 const NodeDef* transpose_node = nullptr; in TEST_F() local1715 const NodeDef* transpose_node = nullptr; in TEST_F() local2174 const NodeDef* transpose_node = node_map.GetNode(optimized_transpose_name); in TEST_F() local
582 auto* transpose_node = context.graph_view->GetNode(transpose_node_name); in TEST_F() local749 auto* transpose_node = context.graph_view->GetNode( in TEST_F() local
2458 bool IsInnerMatrixTransposeNode(const NodeDef& transpose_node, in IsInnerMatrixTransposeNode()
1103 Node* transpose_node = getNodeWithKind(smodule, "aten::transpose"); in TEST() local