Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dlower_tuples.cpp71 block->insertOutput(index + j + 1, input->node()->inputs().at(j)); in flattenTupleInBlockReturn()
92 auto new_block_out = block_node->insertOutput(index + j + 1); in flattenTupleInBlockReturn()
241 n->insertOutput(i + 1 + j)->setType(tt->elements()[j]); in flattenOutputs()
Dloop_unrolling.cpp156 loop->insertOutput(0)->setType(IntType::get()); in replaceLoopCounter()
163 body->insertOutput(1, result); in replaceLoopCounter()
/external/pytorch/torch/csrc/jit/frontend/
Dcanonicalize_modified_loop.cpp48 loop.bodyBlock()->insertOutput(0, new_condition); in canonicalizeModifiedLoop()
Dexit_transforms.cpp216 loop.bodyBlock()->insertOutput(0, new_condition); in transformLoop()
/external/pytorch/test/cpp/jit/
Dtest_shape_analysis.cpp191 if_v.thenBlock()->insertOutput(i, te_group->inputs().at(offset + i)); in TEST()
192 if_v.elseBlock()->insertOutput(i, none_val); in TEST()
193 if_node->insertOutput(i)->setType(OptionalType::create(IntType::get())); in TEST()
/external/pytorch/torch/csrc/jit/ir/
Dir.h598 Value* insertOutput(size_t i);
1085 Value* v = input_->insertOutput(i);
1099 size_t insertOutput(size_t i, Value* n) { in insertOutput() function
Dir.cpp1450 Value* Node::insertOutput(size_t i) { in insertOutput() function in torch::jit::Node
/external/pytorch/torch/csrc/jit/passes/utils/
Dsubgraph_utils.cpp53 Value* new_value = placeholder_node_->insertOutput(i)->copyMetadata( in ValueMapper()