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.cc131 IsIdentityN(node) && !IsIdentityNSingleInput(node); in NumEdgesIfBypassed()
168 const bool is_identity = IsIdentity(node) || IsIdentityNSingleInput(node); in BypassingNodeIsBeneficial()
170 IsIdentityN(node) && !IsIdentityNSingleInput(node); in BypassingNodeIsBeneficial()
229 const bool is_identity = IsIdentity(*node) || IsIdentityNSingleInput(*node); in OptimizeNode()
231 IsIdentityN(*node) && !IsIdentityNSingleInput(*node); in OptimizeNode()
Dpin_to_host_optimizer.cc120 if (IsIdentity(node) || IsIdentityNSingleInput(node)) { in IsNodeOutputPortHostFriendly()
Dmodel_pruner.cc65 if (IsIdentity(node) || IsIdentityNSingleInput(node)) { in IsTrivialOp()
Dconstant_folding.cc255 if (IsIdentity(*output) || IsIdentityNSingleInput(*output)) { in AddControlDependency()
2389 if ((!IsIdentity(*fanout) && !IsIdentityNSingleInput(*fanout)) || in SimplifySwitch()
3013 if ((IsIdentityN(*node) || IsIdentityNSingleInput(*node)) && in PartialConstPropThroughIdentityN()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h83 bool IsIdentityNSingleInput(const NodeDef& node);
Dop_types.cc262 bool IsIdentityNSingleInput(const NodeDef& node) { in IsIdentityNSingleInput() function
Dmutable_graph_view.cc63 if ((IsIdentity(node) || IsIdentityNSingleInput(node)) && in IsIdentityConsumingSwitch()
974 if (IsIdentity(*fanout.node) || IsIdentityNSingleInput(*fanout.node)) { in GetControllingFaninToAdd()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1474 } else if (IsIdentity(node) || IsIdentityNSingleInput(node)) { in MaybeUpdateNodeContextOutput()