Searched refs:ofTensor (Results 1 – 15 of 15) sorted by relevance
| /external/executorch/exir/verification/ |
| D | arg_validator.py | 83 or schema_arg.type == torch.OptionalType.ofTensor()
|
| /external/pytorch/aten/src/ATen/core/ |
| D | union_type.cpp | 20 TypePtr OptionalType::ofTensor() { in ofTensor() function in c10::OptionalType
26 static auto value = ListType::create(OptionalType::ofTensor()); in ofOptionalTensors()
|
| D | jit_type.h | 232 static TypePtr ofTensor();
|
| /external/pytorch/aten/src/ATen/core/dispatch/ |
| D | DispatchKeyExtractor.h | 206 *OptionalType::ofTensor())) { in makeBitsetForDispatchArgs()
|
| /external/pytorch/torch/csrc/jit/runtime/ |
| D | argument_spec.cpp | 36 } else if (typ->isSubtypeOf(*OptionalType::ofTensor())) { in scan()
|
| D | register_prim_ops_fulljit.cpp | 256 c10::impl::GenericList res_list{OptionalType::ofTensor()}; in __anon992a14c81902()
|
| /external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/ |
| D | pattern_conversion.cpp | 325 old_graph->createList(OptionalType::ofTensor(), indices) in ConvertIndexPutToONNX()
|
| /external/pytorch/torch/csrc/jit/frontend/ |
| D | tracer.cpp | 760 g->createList(OptionalType::ofTensor(), fmap(value, getValueTrace))); in addInputs()
774 OptionalType::ofTensor(), fmap(value, getOptTensorValueTrace))); in addInputs()
|
| D | ir_emitter.cpp | 2868 OptionalType::ofTensor(), tensorIndices)) in emitAugAssignmentToSubscript()
2952 OptionalType::ofTensor(), tensorIndices)) in emitSubscriptAssign()
4922 graph->insertNode(graph->createList(OptionalType::ofTensor(), indices)) in emitIndex()
5032 } else if (index->type()->isSubtypeOf(*OptionalType::ofTensor())) { in emitIntAndSliceIndexing()
5106 } else if (expr->type()->isSubtypeOf(*OptionalType::ofTensor())) { in emitIntAndSliceIndexing()
|
| /external/pytorch/torch/csrc/jit/passes/onnx/ |
| D | remove_inplace_ops_for_onnx.cpp | 187 graph->insertNode(graph->createList(OptionalType::ofTensor(), {})) in PrepareCopyForONNX()
|
| /external/pytorch/torch/onnx/ |
| D | utils.py | 530 if output_type.isSubtypeOf(_C.ListType(_C.OptionalType.ofTensor())):
|
| D | symbolic_helper.py | 1493 n.setType(_C.OptionalType.ofTensor())
|
| D | symbolic_opset9.py | 316 n.setType(_C.OptionalType.ofTensor())
|
| /external/pytorch/torch/csrc/jit/python/ |
| D | python_ir.cpp | 1055 .def_static("ofTensor", &OptionalType::ofTensor) in initPythonIRBindings()
|
| /external/pytorch/torch/_C/ |
| D | __init__.pyi.in | 2237 def ofTensor() -> OptionalType: ...
|