Function _export

devtools/intermediate_output_tap/tests/test_tap_pass.py:36–37 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

34
35
36	def _export():
37	return export(_MLP(), (torch.randn(2, 8),), strict=True)
38
39
40	class _RichModel(torch.nn.Module):

test_joint_graphMethod · 0.85

_test_model_with_non_decomp_partitionerMethod · 0.85

test_to_edge_transform_and_lower_with_exceptionMethod · 0.85

exportMethod · 0.85

test_inserts_tap_per_selected_nodeMethod · 0.85

test_appends_user_outputsMethod · 0.85

test_output_indices_contiguous_after_user_outputsMethod · 0.85

test_default_rules_use_stats_on_call_functionsMethod · 0.85

test_mutates_original_and_returns_same_epMethod · 0.85

test_max_tapsMethod · 0.85

test_idempotent_does_not_tap_tapsMethod · 0.85

test_no_match_returns_empty_specsMethod · 0.85

exportFunction · 0.90

_MLPClass · 0.70

no test coverage detected