Searched defs:current_scale (Results 1 – 5 of 5) sorted by relevance
181 __global__ void amp_update_scale_cuda_kernel(float* current_scale, in amp_update_scale_cuda_kernel()223 Tensor& _amp_update_scale_cuda_(Tensor& current_scale, in _amp_update_scale_cuda_()
27 at::Tensor& current_scale, in _amp_update_scale_cpu_()
142 at::Tensor& current_scale, in _amp_update_scale_cpu_kernel()
82 uint32_t current_scale; member
11338 torch::Tensor current_scale = torch::scalar_tensor( in TEST_F() local