Home
last modified time | relevance | path

Searched defs:bfloat16 (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/lib/bfloat16/
Dbfloat16.h45 struct bfloat16 { struct
48 B16_DEVICE_FUNC bfloat16() : value(ZERO_VALUE) {} in bfloat16() function
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() function
69 B16_DEVICE_FUNC explicit bfloat16(const double val) in bfloat16() function
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() argument
82 B16_DEVICE_FUNC explicit bfloat16(const unsigned int val) in bfloat16() argument
85 B16_DEVICE_FUNC explicit bfloat16(const int val) in bfloat16() argument
[all …]
/external/tensorflow/tensorflow/compiler/xla/python/
Dtypes.cc74 py::handle bfloat16(tensorflow::Bfloat16Dtype()); in PrimitiveTypeToDtype() local
Dxla_client.py186 bfloat16 = _xla.bfloat16_dtype() variable
Dxla_client_test.py36 bfloat16 = xla_client.bfloat16 variable
/external/tensorflow/tensorflow/python/lib/core/
Dbfloat16_test.py36 bfloat16 = _pywrap_bfloat16.TF_bfloat16_type() variable
Dbfloat16.cc31 using bfloat16 = Eigen::bfloat16; typedef
/external/tensorflow/tensorflow/python/framework/
Ddtypes.py289 bfloat16 = DType(types_pb2.DT_BFLOAT16) variable
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dquantize_test.cc30 using bfloat16 = tensorflow::bfloat16; typedef
/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1396 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()