↓ 28 callersMethodrunfunc_and_check(
self, inputs_shape, inputs_specs, op_func, with_backward, **kwargs
)
test/auto_parallel/semi_auto_parallel_util.py:80
↓ 27 callersMethodtrain(self, config, model, train_dataloader, criterion, optimizer)
test/auto_parallel/pir/semi_auto_parallel_simple_net_ep.py:360