Searched refs:TF_BFLOAT16 (Results 1 – 6 of 6) sorted by relevance
62 TF_BFLOAT16 = 14, // Float32 truncated to 16 bits. Only for cast ops. enumerator
307 EXPECT_EQ(TF_BFLOAT16, static_cast<TF_DataType>(tensorflow::DT_BFLOAT16)); in TEST()
179 case TF_BFLOAT16: in TF_DataType_to_PyArray_TYPE()
170 *out_tf_datatype = TF_BFLOAT16; in PyArray_TYPE_to_TF_DataType()
55 Bfloat16 DataType = C.TF_BFLOAT16
1173 .value("TF_BFLOAT16", TF_BFLOAT16) in PYBIND11_MODULE()