Home
last modified time | relevance | path

Searched defs:bfloat16 (Results 1 – 5 of 5) 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() function
85 B16_DEVICE_FUNC explicit bfloat16(const int val) in bfloat16() argument
[all …]
/external/tensorflow/tensorflow/python/lib/core/
Dbfloat16_test.py32 bfloat16 = pywrap_tensorflow.TF_bfloat16_type() variable
/external/tensorflow/tensorflow/python/framework/
Ddtypes.py374 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.cc1395 empty_type_wrapper<bfloat16>, const libxsmm_spmdm_handle* handle, in wrapper_libxsmm_spmdm_createSparseSlice_generic_thread()
1405 empty_type_wrapper<bfloat16>, const libxsmm_spmdm_handle* handle, in wrapper_libxsmm_spmdm_compute_generic_thread()