Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dfixup_onnx_controlflow.cpp325 OptionalTypePtr sub_block_input_optional = in FixupONNXLoopNodeInputs() local
330 if (!input->type()->cast<OptionalType>() && sub_block_input_optional) { in FixupONNXLoopNodeInputs()
333 sub_block_input_optional->getElementType(), input->type()); in FixupONNXLoopNodeInputs()
335 sub_block_input_optional = OptionalType::create(merged_type); in FixupONNXLoopNodeInputs()
336 sub_block->inputs().at(i)->setType(sub_block_input_optional); in FixupONNXLoopNodeInputs()
339 Node* opt_node = ONNXOptionalNode(sub_block_input_optional, graph); in FixupONNXLoopNodeInputs()