Searched defs:ArgumentSpec (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | argument_spec.h | 73 struct ArgumentSpec { struct
74 ArgumentSpec(size_t num_flat_tensor_inputs, size_t num_flat_optional_inputs) in ArgumentSpec() function
82 void addOptional(const IValue& input) { in addOptional()
88 void addTensor(const IValue& input, bool with_grad) { in addTensor()
126 bool operator==(const ArgumentSpec& spec) const { argument
144 size_t numTensors() const { in numTensors()
170 // ArgumentSpecCreator takes an initial graph and comes up with a set argument
171 // of simple instructions to compute the ArgumentSpec given a set of argument
|
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | graph_signature.py | 55 ArgumentSpec = Union[ variable
522 def to_input_spec(inp: ArgumentSpec) -> InputSpec:
547 def to_output_spec(idx: int, o: ArgumentSpec) -> OutputSpec:
|
| /aosp_15_r20/packages/inputmethods/LatinIME/native/dicttoolkit/src/utils/ |
| D | arguments_parser.h | 88 ArgumentSpec(const std::string &name, const size_t minCount, const size_t maxCount, in ArgumentSpec() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/java/src/gen/cc/ |
| H A D | op_specs.h | 65 ArgumentSpec(const string& op_def_name, const Variable& var, const Type& type, in ArgumentSpec() function
|
| /aosp_15_r20/external/executorch/exir/serde/ |
| H A D | export_serialize.py | 1053 def serialize_argument_spec(self, x: ep.ArgumentSpec) -> Argument:
|
| /aosp_15_r20/external/pytorch/torch/_export/serde/ |
| H A D | serialize.py | 1083 def serialize_argument_spec(self, x: ep.ArgumentSpec) -> Argument:
|