Home
last modified time | relevance | path

Searched defs:float8_e5m2 (Results 1 – 5 of 5) sorted by relevance

/external/pytorch/torch/_inductor/codegen/
Dcpp_prefix.h48 typedef at::Float8_e5m2 float8_e5m2; typedef
/external/ml_dtypes/ml_dtypes/include/
Dfloat8.h54 class float8_e5m2; variable
377 explicit EIGEN_DEVICE_FUNC float8_e5m2(T f8) : float8_e5m2(ConvertFrom(f8)) {} in float8_e5m2() function
/external/ml_dtypes/ml_dtypes/tests/
Dcustom_float_test.py42 float8_e5m2 = ml_dtypes.float8_e5m2 variable
/external/pytorch/torch/
Dstorage.py329 def float8_e5m2(self): member in _StorageBase
1359 def float8_e5m2(self): member in TypedStorage
/external/pytorch/torch/csrc/inductor/aoti_torch/
Dshim_common.cpp131 AOTI_TORCH_DTYPE_IMPL(float8_e5m2, Float8_e5m2) in AOTI_TORCH_DTYPE_IMPL() argument