Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util.h240 bool IsStackOp(const Node& node) const { in IsStackOp() function
Dcompilability_check_util.cc506 if (!op_filter_.allow_stack_ops && IsStackOp(node)) { in IsCompilableNode()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h183 bool IsStackOp(const NodeDef& node);
Dop_types.cc551 bool IsStackOp(const NodeDef& node) { in IsStackOp() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc500 } else if (IsStackOp(fanout_node) || IsStackCloseOp(fanout_node) || in GetStackPushNodesToConvert()
525 if (IsStackOp(optimized_graph->node(node_idx))) { in RemoveStackOps()