Home
last modified time | relevance | path

Searched refs:B16_DEVICE_FUNC (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/lib/bfloat16/
Dbfloat16.h26 #define B16_DEVICE_FUNC __host__ __device__ macro
29 #define B16_DEVICE_FUNC macro
48 B16_DEVICE_FUNC bfloat16() : value(ZERO_VALUE) {} in bfloat16()
50 B16_DEVICE_FUNC static bfloat16 truncate_to_bfloat16(const float v) { in truncate_to_bfloat16()
65 B16_DEVICE_FUNC explicit bfloat16(const float v) { in bfloat16()
69 B16_DEVICE_FUNC explicit bfloat16(const double val) in bfloat16()
73 B16_DEVICE_FUNC explicit bfloat16(const complex64& val) in bfloat16()
76 B16_DEVICE_FUNC explicit bfloat16(const complex128& val) in bfloat16()
79 B16_DEVICE_FUNC explicit bfloat16(const unsigned short val) in bfloat16()
82 B16_DEVICE_FUNC explicit bfloat16(const unsigned int val) in bfloat16()
[all …]
Dbfloat16.cc25 B16_DEVICE_FUNC bfloat16::operator Eigen::half() const { in operator Eigen::half()