Searched refs:perm_node (Results 1 – 2 of 2) sorted by relevance
1115 const Node* perm_node = e->src(); in CheckForTranspose() local1118 if (perm_node->type_string() == "Const" && in CheckForTranspose()1122 TF_CHECK_OK(GetNodeAttr(perm_node->def(), "value", &proto)); in CheckForTranspose()1125 GetNodeAttr(perm_node->def(), "dtype", &type); in CheckForTranspose()
2257 const NodeDef* perm_node = node_map->GetNode(transpose_node.input(1)); in IsInnerMatrixTransposeNode() local2259 if (ValuesFromConstNode(*perm_node, &perm32)) { in IsInnerMatrixTransposeNode()2263 if (ValuesFromConstNode(*perm_node, &perm64)) { in IsInnerMatrixTransposeNode()