MCPcopy Create free account
hub / github.com/pytorch/executorch / OutputSpec

Class OutputSpec

exir/serde/schema.py:326–333  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

324
325@dataclass(repr=False)
326class OutputSpec(_Union):
327 user_output: UserOutputSpec
328 loss_output: LossOutputSpec
329 buffer_mutation: BufferMutationSpec
330 gradient_to_parameter: GradientToParameterSpec
331 gradient_to_user_input: GradientToUserInputSpec
332 user_input_mutation: UserInputMutationSpec
333 token: OutputTokenSpec
334
335
336@dataclass

Callers 11

programMethod · 0.85
_get_new_signatureFunction · 0.85
weights_to_outputs_passFunction · 0.85
captureFunction · 0.85
outputMethod · 0.85
create_mutable_bufferFunction · 0.85

Calls

no outgoing calls