Class SimpleModule

backends/mlx/test/test_pattern_utils.py:170–172 · view source on GitHub ↗

Source from the content-addressed store, hash-verified

168	from executorch.backends.mlx.pattern_utils import OpStep
169
170	class SimpleModule(torch.nn.Module):
171	def forward(self, x):
172	return torch.rsqrt(x)
173
174	graph = get_exported_graph(SimpleModule(), (torch.randn(4, 4),))
175	rsqrt_node = find_node_by_target(graph, "rsqrt")

test_match_target_basicMethod · 0.70

test_match_target_non_call_functionMethod · 0.70

test_has_usersMethod · 0.70

test_no_users_after_removalMethod · 0.70

test_matches_with_opMethod · 0.70

test_matches_with_predicateMethod · 0.70

test_matches_no_op_no_predicateMethod · 0.70

test_matches_nargs_intMethod · 0.70

test_matches_nargs_tupleMethod · 0.70

test_matches_kwargs_emptyMethod · 0.70

test_matches_arg_indexMethod · 0.70

test_walk_back_single_stepMethod · 0.70

no outgoing calls

test_match_target_basicMethod · 0.56

test_match_target_non_call_functionMethod · 0.56

test_has_usersMethod · 0.56

test_no_users_after_removalMethod · 0.56

test_matches_with_opMethod · 0.56

test_matches_with_predicateMethod · 0.56

test_matches_no_op_no_predicateMethod · 0.56

test_matches_nargs_intMethod · 0.56

test_matches_nargs_tupleMethod · 0.56

test_matches_kwargs_emptyMethod · 0.56

test_matches_arg_indexMethod · 0.56

test_walk_back_single_stepMethod · 0.56