Searched refs:perm_node (Results 1 – 2 of 2) sorted by relevance
1408 const Node* perm_node = e->src(); in CheckForTranspose() local1411 if (perm_node->type_string() == "Const" && in CheckForTranspose()1415 TF_CHECK_OK(GetNodeAttr(perm_node->def(), "value", &proto)); in CheckForTranspose()1418 TF_CHECK_OK(GetNodeAttr(perm_node->def(), "dtype", &type)); in CheckForTranspose()
2371 const NodeDef* perm_node = node_map->GetNode(transpose_node.input(1)); in IsInnerMatrixTransposeNode() local2373 if (ValuesFromConstNode(*perm_node, &perm32)) { in IsInnerMatrixTransposeNode()2377 if (ValuesFromConstNode(*perm_node, &perm64)) { in IsInnerMatrixTransposeNode()