Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dlower_tuples.cpp44 auto new_construct_node = in flattenTupleInLoopParams() local
48 new_construct_node->addInput(new_block_in); in flattenTupleInLoopParams()
51 new_construct_node->output()->setType(block->inputs().at(index - 1)->type()); in flattenTupleInLoopParams()
52 new_construct_node->copyMetadata(n); in flattenTupleInLoopParams()
54 new_construct_node->output()); in flattenTupleInLoopParams()
65 Node* new_construct_node = nullptr; in flattenTupleInBlockReturn() local
89 new_construct_node = block->owningGraph()->create(prim::TupleConstruct); in flattenTupleInBlockReturn()
90 new_construct_node->insertAfter(block_node); in flattenTupleInBlockReturn()
93 new_construct_node->addInput(new_block_out); in flattenTupleInBlockReturn()
96 new_construct_node->output()->setType(tuple_output->type()); in flattenTupleInBlockReturn()
[all …]