Home
last modified time | relevance | path

Searched defs:enter_node (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dcheck_strict_fusion.cpp30 Node* enter_node) { in collectValuesUsedInGuard()
59 static void checkForUnfusedOps(Node* enter_node) { in checkForUnfusedOps()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dlower_while_op_test.cc216 const Node* enter_node = nullptr; in TEST() local
337 const Node* enter_node = nullptr; in TEST() local
H A Dlower_while_op.cc245 Node* enter_node; in CreateEnterNodes() local
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dcollective_order.cc40 bool enter_node = in DiscoverDataDependencies() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/framework/
H A Dgradients.cc429 Node* enter_node = while_ctx->enter_nodes()[i]; in ProcessWhileLoop() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dhost_training_loop_optimization_util.cc95 const Node* enter_node, const absl::flat_hash_set<Node*> execute_nodes) { in ResourceOnlyUsedForTPUExecuteInLoop()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dloop_optimizer.cc662 NodeDef* enter_node = nullptr; in CheckForDeadFanout() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dencapsulate_subgraphs_pass.cc999 Node* enter_node = options.FinalizeBuilder(&enter_builder); in AddDummyShapedNode() local