Searched refs:tuple_out (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/_subclasses/ |
| D | schema_check_mode.py | 161 tuple_out = out if isinstance(out, tuple) else (out,)
162 tuple_out = tree_map(unwrap, tuple_out)
175 for j in range(len(tuple_out)):
179 has_aliased(tuple_out[j], after)
193 if after is tuple_out[j] and isinstance(after, torch.Tensor):
223 if has_aliased(tuple_out[i], tuple_out[j]):
|
| /external/pytorch/test/export/ |
| D | opinfo_schema.py | 55 tuple_out = out if isinstance(out, tuple) else (out,)
56 tuple_out = tree_map(unwrap, tuple_out)
60 for j in range(len(tuple_out)):
|
| /external/pytorch/torch/csrc/jit/passes/ |
| D | lower_tuples.cpp | 146 auto tuple_out = graph->createTuple(values); in removeTupleNodes() local
147 tuple_out->copyMetadata(n); in removeTupleNodes()
149 graph->insertNode(tuple_out); in removeTupleNodes()
150 n->output()->replaceAllUsesWith(tuple_out->output()); in removeTupleNodes()
|
| /external/pytorch/test/onnx/ |
| D | test_pytorch_onnx_onnxruntime.py | 509 tuple_out: Optional[Tuple[Tensor]] = None variable in TestONNXRuntime.test_dict_output.DictModelOutput
516 tuple_out=(b, c),
|