Searched defs:ModelClass (Results 1 – 3 of 3) sorted by relevance
81 def test_tracer(self, ModelClass): argument146 def test_tracer_kwargs(self, ModelClass): argument
82 def test_model_split(self, ModelClass): argument
205 def test_grad_with_tracer(self, ScheduleClass, ModelClass): argument