Searched defs:bfloat16 (Results 1 – 9 of 9) sorted by relevance
45 struct bfloat16 { struct48 B16_DEVICE_FUNC bfloat16() : value(ZERO_VALUE) {} in bfloat16() function50 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() argument76 B16_DEVICE_FUNC explicit bfloat16(const complex128& val) in bfloat16() argument79 B16_DEVICE_FUNC explicit bfloat16(const unsigned short val) in bfloat16() argument82 B16_DEVICE_FUNC explicit bfloat16(const unsigned int val) in bfloat16() argument85 B16_DEVICE_FUNC explicit bfloat16(const int val) in bfloat16() argument[all …]
74 py::handle bfloat16(tensorflow::Bfloat16Dtype()); in PrimitiveTypeToDtype() local
186 bfloat16 = _xla.bfloat16_dtype() variable
36 bfloat16 = xla_client.bfloat16 variable
36 bfloat16 = _pywrap_bfloat16.TF_bfloat16_type() variable
31 using bfloat16 = Eigen::bfloat16; typedef
289 bfloat16 = DType(types_pb2.DT_BFLOAT16) variable
30 using bfloat16 = tensorflow::bfloat16; typedef
1396 empty_type_wrapper<bfloat16>, const libxsmm_spmdm_handle* handle, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()1406 empty_type_wrapper<bfloat16>, const libxsmm_spmdm_handle* handle, in wrapper_libxsmm_spmdm_compute_generic_thread()