Home
last modified time | relevance | path

Searched refs:IsStackOp (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dop_types.h161 bool IsStackOp(const NodeDef& node);
Dop_types.cc472 bool IsStackOp(const NodeDef& node) { in IsStackOp() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc501 } else if (IsStackOp(fanout_node) || IsStackCloseOp(fanout_node) || in GetStackPushNodesToConvert()
526 if (IsStackOp(optimized_graph->node(node_idx))) { in RemoveStackOps()