Searched refs:TF_BFLOAT16 (Results 1 – 7 of 7) sorted by relevance
62 TF_BFLOAT16 = 14, // Float32 truncated to 16 bits. Only for cast ops. enumerator
310 EXPECT_EQ(TF_BFLOAT16, static_cast<TF_DataType>(tensorflow::DT_BFLOAT16)); in TEST()
97 case TF_DataType::TF_BFLOAT16: in GetDlDataType()209 *tf_dtype = TF_DataType::TF_BFLOAT16; in TfDataTypeFormDlDataType()
179 case TF_BFLOAT16: in TF_DataType_to_PyArray_TYPE()
172 *out_tf_datatype = TF_BFLOAT16; in PyArray_TYPE_to_TF_DataType()
62 Bfloat16 DataType = C.TF_BFLOAT16
1197 .value("TF_BFLOAT16", TF_BFLOAT16) in PYBIND11_MODULE()