Searched defs:OutputSpec (Results 1 – 10 of 10) sorted by relevance
36 struct TORCH_API OutputSpec { struct48 std::vector<int64_t> sizes_;50 std::optional<double> qscale_;51 std::optional<int64_t> qzero_;
57 OutputSpec::OutputSpec(const c10::IValue& value) { in OutputSpec() function in torch::jit::mobile::nnc::OutputSpec
175 const auto &OutputSpec = OutputSpecs[I]; in TFModelEvaluatorImpl() local
848 output_specs_to_delete: Dict[str, OutputSpec], argument
315 class OutputSpec(_Union): class
1021 def serialize_output_spec(self, spec: ep.OutputSpec) -> OutputSpec:1790 def deserialize_output_spec(self, o: OutputSpec) -> ep.OutputSpec:
322 class OutputSpec(_Union): class
991 def serialize_output_spec(self, spec: ep.OutputSpec) -> OutputSpec:1767 def deserialize_output_spec(self, o: OutputSpec) -> ep.OutputSpec:
108 class OutputSpec: class
572 def specs_dict(specs: list[InputSpec | OutputSpec], title: str):