Home
last modified time | relevance | path

Searched defs:exit_node (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state.cc506 const NodeDef& exit_node, in FindLoopCondFromExitNode()
571 for (const NodeDef* exit_node : exit_nodes) { in PruneGraphDef() local
640 for (const NodeDef* exit_node : loop_cond_to_exit_nodes[node]) { in PruneGraphDef() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_while_op.cc393 Node* exit_node; in CreateExitNodes() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/framework/
H A Dgradients.cc391 Status SymbolicGradientBuilder::ProcessWhileLoop(Node* exit_node, in ProcessWhileLoop()
/aosp_15_r20/external/pytorch/torch/csrc/jit/frontend/
H A Dexit_transforms.cpp743 for (auto exit_node : exit_block->nodes()) { in convertWithBlocksToEnterExitNodes() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dpfor.py211 def __init__(self, exit_node, pfor_ops, fallback_to_while_loop, pfor_config): argument