Searched refs:TF_BFLOAT16 (Results 1 – 5 of 5) sorted by relevance
178 case TF_BFLOAT16: in TF_DataType_to_PyArray_TYPE()
130 *out_tf_datatype = TF_BFLOAT16; in PyArray_TYPE_to_TF_DataType()
55 Bfloat16 DataType = C.TF_BFLOAT16
113 TF_BFLOAT16 = 14, // Float32 truncated to 16 bits. Only for cast ops. enumerator
286 EXPECT_EQ(TF_BFLOAT16, static_cast<TF_DataType>(tensorflow::DT_BFLOAT16)); in TEST()