Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h160 bool IsSqueeze(const NodeDef& node);
Dop_types.cc470 bool IsSqueeze(const NodeDef& node) { return node.op() == "Squeeze"; } in IsSqueeze() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc306 if (!squeeze.node || !IsSqueeze(*squeeze.node) || in FindConv2DWithSqueezeAndBias()
Dlayout_optimizer.cc2100 } else if (IsSqueeze(*node)) { in Expand()
Dconstant_folding.cc2273 if (use_shape_info && IsSqueeze(*node) && in SimplifySqueeze()