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

Method get_export_kwargs

test/models/export_program.py:47–52  ·  view source on GitHub ↗

Returns custom trace params for ExportedModule.

()

Source from the content-addressed store, hash-verified

45
46 @staticmethod
47 def get_export_kwargs() -> Dict[str, Any]:
48 """Returns custom trace params for ExportedModule."""
49 return {
50 # aten::max.default does not have an out variant.
51 "ignore_to_out_var_failure": True,
52 }
53
54
55class ModuleIndex(nn.Module):

Callers 1

export_module_to_programFunction · 0.45

Calls

no outgoing calls

Tested by

no test coverage detected