Searched refs:tensor_to_scale (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_matmul_cuda.py | 255 def tensor_to_scale(x: torch.Tensor, float8_dtype: torch.dtype, dim=None): function 393 x_scale = tensor_to_scale(x, input_dtype).float() 394 y_scale = tensor_to_scale(y, input_dtype).float() 419 out_scaled_mm = out_scaled_mm / tensor_to_scale(out_scaled_mm, input_dtype) 422 out_emulated = out_emulated / tensor_to_scale(out_emulated, input_dtype) 442 x_scale = tensor_to_scale(x, input_dtype).float() 443 y_scale = tensor_to_scale(y, input_dtype).float() 468 out_scaled_mm = out_scaled_mm / tensor_to_scale(out_scaled_mm, input_dtype) 471 out_emulated = out_emulated / tensor_to_scale(out_emulated, input_dtype) 679 x_scales = tensor_to_scale(x, input_dtype, dim=1).float() [all …]
|