Searched refs:backward_api_called (Results 1 – 1 of 1) sorted by relevance
243 bool backward_api_called = accumulate_grad; in THPEngine_run_backward() local245 !backward_api_called || at::impl::VmapMode::current_vmap_level() == 0, in THPEngine_run_backward()378 if (!backward_api_called && inputs != nullptr) { in THPEngine_run_backward()