Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/_inductor/codegen/
H A Dcpp_prefix.h48 typedef at::Float8_e5m2 float8_e5m2; typedef
/aosp_15_r20/external/pytorch/torch/
H A Dstorage.py329 def float8_e5m2(self): member in _StorageBase
1359 def float8_e5m2(self): member in TypedStorage
/aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/
H A Dshim_common.cpp131 AOTI_TORCH_DTYPE_IMPL(float8_e5m2, Float8_e5m2) in AOTI_TORCH_DTYPE_IMPL() argument