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 …]
|