Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc153 IsIdentityN(node) && !IsIdentityNSingleInput(node); in NumEdgesIfBypassed()
190 const bool is_identity = IsIdentity(node) || IsIdentityNSingleInput(node); in BypassingNodeIsBeneficial()
192 IsIdentityN(node) && !IsIdentityNSingleInput(node); in BypassingNodeIsBeneficial()
251 const bool is_identity = IsIdentity(*node) || IsIdentityNSingleInput(*node); in OptimizeNode()
253 IsIdentityN(*node) && !IsIdentityNSingleInput(*node); in OptimizeNode()
Dpin_to_host_optimizer.cc124 if (IsIdentity(node) || IsIdentityNSingleInput(node)) { in IsNodeOutputPortHostFriendly()
Dmodel_pruner.cc66 if (IsIdentity(node) || IsIdentityNSingleInput(node)) { in IsTrivialOp()
Dconstant_folding.cc226 if (IsIdentity(*output) || IsIdentityNSingleInput(*output)) { in AddControlDependency()
2649 if ((!IsIdentity(*fanout) && !IsIdentityNSingleInput(*fanout)) || in SimplifySwitch()
3561 if (!(IsIdentityN(*node) || IsIdentityNSingleInput(*node)) || in PartialConstPropThroughIdentityN()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h97 bool IsIdentityNSingleInput(const NodeDef& node);
Dop_types.cc323 bool IsIdentityNSingleInput(const NodeDef& node) { in IsIdentityNSingleInput() function
Dmutable_graph_view.cc63 if ((IsIdentity(node) || IsIdentityNSingleInput(node)) && in IsIdentityConsumingSwitch()
977 if (IsIdentity(*fanout.node) || IsIdentityNSingleInput(*fanout.node)) { in GetControllingFaninToAdd()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1773 } else if (IsIdentity(node) || IsIdentityNSingleInput(node)) { in MaybeUpdateNodeContextOutput()