Searched refs:perm_node (Results 1 – 2 of 2) sorted by relevance
1352 const Node* perm_node = e->src(); in CheckForTranspose() local1355 if (perm_node->type_string() == "Const" && in CheckForTranspose()1359 TF_CHECK_OK(GetNodeAttr(perm_node->def(), "value", &proto)); in CheckForTranspose()1362 TF_CHECK_OK(GetNodeAttr(perm_node->def(), "dtype", &type)); in CheckForTranspose()
2464 const NodeDef* perm_node = node_map->GetNode(transpose_node.input(1)); in IsInnerMatrixTransposeNode() local2466 if (ValuesFromConstNode(*perm_node, &perm32)) { in IsInnerMatrixTransposeNode()2470 if (ValuesFromConstNode(*perm_node, &perm64)) { in IsInnerMatrixTransposeNode()