Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset16.py160 utils._add_output_to_block(if_context.block, self_reshape)
162 utils._add_output_to_block(if_context.block, index_reshape)
164 utils._add_output_to_block(if_context.block, src_reshape)
167 utils._add_output_to_block(else_context.block, self_identity)
169 utils._add_output_to_block(else_context.block, index_identitye)
171 utils._add_output_to_block(else_context.block, src_identity)
180 utils._add_output_to_block(if_context.block, result_squeezed)
182 utils._add_output_to_block(else_context.block, result_identity)
H A Dsymbolic_opset13.py232 utils._add_output_to_block(loop_block, cond_out)
233 utils._add_output_to_block(loop_block, final_splits)
548 utils._add_output_to_block(if_context_greater.block, dims_)
550 utils._add_output_to_block(else_context_greater.block, identity_dim)
573 utils._add_output_to_block(if_context_less.block, self_)
575 utils._add_output_to_block(else_context_less.block, identity_self)
715 utils._add_output_to_block(loop_block, cond_out)
716 utils._add_output_to_block(loop_block, final_splits)
838 utils._add_output_to_block(if_context.block, final_non_overrun)
839 utils._add_output_to_block(else_context.block, final_overrun)
H A Dsymbolic_opset12.py368 utils._add_output_to_block(loop_block, cond_out)
369 utils._add_output_to_block(loop_block, concat)
H A Dsymbolic_opset11.py891 utils._add_output_to_block(if_context.block, squeeze_)
893 utils._add_output_to_block(else_context.block, identity_)
1454 utils._add_output_to_block(if_context_equal.block, result_if)
1458 utils._add_output_to_block(else_context_equal.block, result_else)
H A Dsymbolic_helper.py2038 utils._add_output_to_block(loop_block, cond_out)
2039 utils._add_output_to_block(loop_block, embeddings)
H A Dutils.py1732 def _add_output_to_block(block: _C.Block, value: _C.Value) -> int: function
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_pytorch_onnx_shape_inference.py358 utils._add_output_to_block(if_context.block, block1_output)
359 utils._add_output_to_block(else_context.block, block2_output)