Searched refs:TwoLayerNetModule (Results 1 – 1 of 1) sorted by relevance
23 class TwoLayerNetModule(torch.nn.Module): class76 self.linear_test(TwoLayerNetModule)80 self.linear_test(TwoLayerNetModule, profiler_output_path=fname)