Searched defs:bfloat16 (Results 1 – 12 of 12) sorted by relevance
45 struct bfloat16 { struct48 B16_DEVICE_FUNC bfloat16() : value(ZERO_VALUE) {} in bfloat16() argument50 B16_DEVICE_FUNC static bfloat16 truncate_to_bfloat16(const float v) { in truncate_to_bfloat16() argument65 B16_DEVICE_FUNC explicit bfloat16(const float v) { in bfloat16() function69 B16_DEVICE_FUNC explicit bfloat16(const double val) in bfloat16() function73 B16_DEVICE_FUNC explicit bfloat16(const complex64& val) in bfloat16() function76 B16_DEVICE_FUNC explicit bfloat16(const complex128& val) in bfloat16() argument79 B16_DEVICE_FUNC explicit bfloat16(const unsigned short val) in bfloat16() function82 B16_DEVICE_FUNC explicit bfloat16(const unsigned int val) in bfloat16() function85 B16_DEVICE_FUNC explicit bfloat16(const int val) in bfloat16() argument[all …]
58 struct bfloat16 : public bfloat16_impl::bfloat16_base { struct60 typedef bfloat16_impl::__bfloat16_raw __bfloat16_raw; argument62 EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16() {} in bfloat16() function64 …EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16(const __bfloat16_raw& h) : bfloat16_impl::bfloat16_base… in bfloat16() argument66 explicit EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16(bool b) in bfloat16() argument70 explicit EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16(T val) in bfloat16() argument73 explicit EIGEN_DEVICE_FUNC bfloat16(float f) in bfloat16() argument79 explicit EIGEN_DEVICE_FUNC EIGEN_CONSTEXPR bfloat16(const std::complex<RealScalar>& val) in bfloat16() function
84 bfloat16() in bfloat16() function92 explicit bfloat16(float v) in bfloat16() function
30 using bfloat16 = arm_compute::bfloat16; typedef
74 py::handle bfloat16(tensorflow::Bfloat16Dtype()); in PrimitiveTypeToDtype() local
193 bfloat16 = _xla.bfloat16_dtype() variable
45 bfloat16 = xla_client.bfloat16 variable
36 bfloat16 = _pywrap_bfloat16.TF_bfloat16_type() variable
31 using bfloat16 = Eigen::bfloat16; typedef
361 bfloat16 = DType(types_pb2.DT_BFLOAT16) variable
30 using bfloat16 = tensorflow::bfloat16; typedef
1400 empty_type_wrapper<bfloat16>, const libxsmm_spmdm_handle* handle, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()1410 empty_type_wrapper<bfloat16>, const libxsmm_spmdm_handle* handle, in wrapper_libxsmm_spmdm_compute_generic_thread()