Searched defs:growth_tracker (Results 1 – 4 of 4) sorted by relevance
182 int* growth_tracker, in amp_update_scale_cuda_kernel()224 Tensor& growth_tracker, in _amp_update_scale_cuda_()
28 at::Tensor& growth_tracker, in _amp_update_scale_cpu_()
143 at::Tensor& growth_tracker, in _amp_update_scale_cpu_kernel()
11336 torch::Tensor growth_tracker = torch::scalar_tensor( in TEST_F() local