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.cc1131 NodeDef* node_perm; in TrySimplify() local
1132 TF_RETURN_IF_ERROR(GetInputNode(node->input(1), &node_perm)); in TrySimplify()
1133 if (!IsConstant(*node_perm)) { in TrySimplify()
1137 TF_RETURN_IF_ERROR(GetPermutation(*node_perm, &node_perm_values)); in TrySimplify()
1185 Status GetPermutation(const NodeDef& node_perm, in GetPermutation() argument
1188 if (ValuesFromConstNode(node_perm, &perm32)) { in GetPermutation()
1195 if (ValuesFromConstNode(node_perm, perm64)) { in GetPermutation()
1199 node_perm.name()); in GetPermutation()