Home
last modified time | relevance | path

Searched refs:outer_node (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dbailout_graph.cpp98 auto outer_node = n->owningBlock()->owningNode(); in buildBailOutBlockFrom() local
99 if (outer_node) { in buildBailOutBlockFrom()
100 if (outer_node->kind() == prim::Loop) { in buildBailOutBlockFrom()
101 buildBailOutLoop(outer_node); in buildBailOutBlockFrom()
102 } else if (outer_node->kind() == prim::If) { in buildBailOutBlockFrom()
103 buildBailOutIf(b->outputs(), outer_node); in buildBailOutBlockFrom()
120 void buildBailOutLoop(Node* outer_node) { in buildBailOutLoop()
121 LoopView lv(outer_node); in buildBailOutLoop()
157 ->setSourceRange(outer_node->sourceRange()); in buildBailOutLoop()
183 if (outer_node->next()) { in buildBailOutLoop()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Darithmetic_optimizer_test.cc2714 const NodeDef* outer_node = node_map.GetNode(outer_add_name); in TEST_F() local
2715 ASSERT_NE(outer_node, nullptr); in TEST_F()
2716 EXPECT_EQ(outer_node->op(), "AddV2"); in TEST_F()
2717 ASSERT_EQ(outer_node->input_size(), 2); in TEST_F()
2718 EXPECT_EQ(outer_node->input(0), outer_0_add_name); in TEST_F()
2719 EXPECT_EQ(outer_node->input(1), inner_2_add_name); in TEST_F()