| /aosp_15_r20/external/tensorflow/tensorflow/core/lib/bfloat16/ |
| H A D | bfloat16.h | 45 struct bfloat16 { struct
48 B16_DEVICE_FUNC bfloat16() : value(ZERO_VALUE) {} in bfloat16() argument
50 B16_DEVICE_FUNC static bfloat16 truncate_to_bfloat16(const float v) { in truncate_to_bfloat16() argument
65 B16_DEVICE_FUNC explicit bfloat16(const float v) { in bfloat16() argument
69 B16_DEVICE_FUNC explicit bfloat16(const double val) in bfloat16() argument
73 B16_DEVICE_FUNC explicit bfloat16(const complex64& val) in bfloat16() argument
76 B16_DEVICE_FUNC explicit bfloat16(const complex128& val) in bfloat16() argument
79 B16_DEVICE_FUNC explicit bfloat16(const unsigned short val) in bfloat16() function
82 B16_DEVICE_FUNC explicit bfloat16(const unsigned int val) in bfloat16() function
85 B16_DEVICE_FUNC explicit bfloat16(const int val) in bfloat16() argument
[all …]
|
| /aosp_15_r20/external/eigen/Eigen/src/Core/arch/Default/ |
| H A D | BFloat16.h | 58 struct bfloat16 : public bfloat16_impl::bfloat16_base { struct
60 typedef bfloat16_impl::__bfloat16_raw __bfloat16_raw; argument
62 EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16() {} in bfloat16() function
64 …EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16(const __bfloat16_raw& h) : bfloat16_impl::bfloat16_base… in bfloat16() function
66 explicit EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16(bool b) in bfloat16() argument
70 explicit EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16(T val) in bfloat16() function
73 explicit EIGEN_DEVICE_FUNC bfloat16(float f) in bfloat16() function
79 explicit EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16(const std::complex<RealScalar>& val) in bfloat16() function
|
| /aosp_15_r20/external/ComputeLibrary/support/ |
| H A D | Bfloat16.h | 85 bfloat16() in bfloat16() function
93 bfloat16(float v) in bfloat16() function
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/ |
| H A D | bfloat.hpp | 30 using bfloat16 = arm_compute::bfloat16; typedef
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_prefix.h | 45 typedef at::BFloat16 bfloat16; typedef
|
| /aosp_15_r20/external/pytorch/torch/_refs/ |
| H A D | _conversions.py | 50 bfloat16 = _make_conversion_method("bfloat16", torch.bfloat16) variable
|
| /aosp_15_r20/external/pytorch/c10/util/ |
| H A D | BFloat16-inl.h | 84 inline C10_HOST_DEVICE BFloat16::operator sycl::ext::oneapi::bfloat16() const { in bfloat16() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | types.cc | 75 py::handle bfloat16(tensorflow::Bfloat16Dtype()); in PrimitiveTypeToDtype() local
|
| H A D | xla_client.py | 138 bfloat16 = _xla.bfloat16_dtype() variable
|
| H A D | xla_client_test.py | 46 bfloat16 = xla_client.bfloat16 variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | dtypes.py | 398 bfloat16 = DType(types_pb2.DT_BFLOAT16) variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | quantize_test.cc | 30 using bfloat16 = tensorflow::bfloat16; typedef
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | storage.py | 317 def bfloat16(self): member in _StorageBase
1344 def bfloat16(self): member in TypedStorage
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/lib/core/ |
| H A D | bfloat16_test.py | 34 bfloat16 = _pywrap_bfloat16.TF_bfloat16_type() variable
|
| H A D | bfloat16.cc | 1730 using bfloat16 = Eigen::bfloat16; typedef
|
| /aosp_15_r20/external/pytorch/torch/distributed/nn/api/ |
| H A D | remote_module.py | 358 def bfloat16(self: T) -> T: # type: ignore[return] member in _RemoteModule
|
| /aosp_15_r20/external/pytorch/torch/_subclasses/ |
| H A D | functional_tensor.py | 296 bfloat16 = _conversion_method_template(dtype=torch.bfloat16) variable in FunctionalTensor
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | sparse_matmul_op.cc | 1405 empty_type_wrapper<bfloat16>, const libxsmm_spmdm_handle* handle, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
1415 empty_type_wrapper<bfloat16>, const libxsmm_spmdm_handle* handle, in wrapper_libxsmm_spmdm_compute_generic_thread()
|
| /aosp_15_r20/external/pytorch/torch/nn/modules/ |
| H A D | module.py | 1167 def bfloat16(self: T) -> T: member in Module
|