Searched refs:make_linear_module (Results 1 – 1 of 1) sorted by relevance
636 def make_linear_module(output_size, name): function645 linear1 = make_linear_module(output_size=2, name="foo")660 linear2 = make_linear_module(output_size=2, name="foo")