Home
last modified time | relevance | path

Searched refs:node_perm (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc1103 NodeDef* node_perm; in TrySimplify() local
1104 TF_RETURN_IF_ERROR(GetInputNode(node->input(1), &node_perm)); in TrySimplify()
1105 if (!IsConstant(*node_perm)) { in TrySimplify()
1109 TF_RETURN_IF_ERROR(GetPermutation(*node_perm, &node_perm_values)); in TrySimplify()
1145 Status GetPermutation(const NodeDef& node_perm, in GetPermutation() argument
1148 if (ValuesFromConstNode(node_perm, &perm32)) { in GetPermutation()
1155 if (ValuesFromConstNode(node_perm, perm64)) { in GetPermutation()
1159 node_perm.name()); in GetPermutation()