Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dshape_optimizer.cc42 if (!IsShape(node)) { in Optimize()
Dlayout_optimizer.cc2094 } else if (IsShape(*node) || IsShapeN(*node)) { in Expand()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h146 bool IsShape(const NodeDef& node);
Dop_types.cc436 bool IsShape(const NodeDef& node) { return node.op() == "Shape"; } in IsShape() function
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.cc1417 } else if (IsShape(node)) { in MaybeUpdateNodeContextOutput()