Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/
H A Dpromoted_prim_ops.cpp142 int elem_ty_val = 0; in toList() local
146 pop(stack, elem_ty_val); in toList()
158 if (elem_ty_val == 0) { in toList()
160 } else if (elem_ty_val == 1) { in toList()
162 } else if (elem_ty_val == 2) { in toList()
164 } else if (elem_ty_val == 3) { in toList()
/aosp_15_r20/external/pytorch/torch/csrc/jit/ir/
H A Dir.cpp1933 Value* elem_ty_val = insertConstant(IValue(elem_ty)); in insertToList() local
1934 Node* n = insertNode(create(prim::tolist, {v, dim_val, elem_ty_val})); in insertToList()
/aosp_15_r20/external/pytorch/torch/onnx/
H A Dsymbolic_opset9.py6355 def prim_tolist(g: jit_utils.GraphContext, input, dim_val, elem_ty_val): argument