Returns custom trace params for ExportedModule.
()
| 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 | |
| 55 | class ModuleIndex(nn.Module): |
no outgoing calls
no test coverage detected