Searched refs:TF_HALF (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 76 return TF_HALF; in GetTensorFlowDataType()
114 case TF_HALF: in GetTensorFlowLiteType()
|
| D | util_test.cc | 119 EXPECT_EQ(TF_HALF, GetTensorFlowDataType(kTfLiteFloat16)); in TEST()
135 EXPECT_EQ(kTfLiteFloat16, GetTensorFlowLiteType(TF_HALF)); in TEST()
|
| /external/tensorflow/tensorflow/c/ |
| D | tf_datatype.h | 67 TF_HALF = 19, enumerator
|
| D | kernels_test.cc | 714 const TF_DataType list[] = {TF_FLOAT, TF_DOUBLE, TF_HALF, TF_COMPLEX128}; in TEST_F()
894 TEST_KERNEL_TYPE_CONSTRAINT(TF_HALF, DT_HALF);
|
| D | c_api_test.cc | 318 EXPECT_EQ(TF_HALF, static_cast<TF_DataType>(tensorflow::DT_HALF)); in TEST()
2194 const TF_DataType list[] = {TF_FLOAT, TF_DOUBLE, TF_HALF, TF_COMPLEX128}; in TEST_F()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | dlpack.cc | 81 case TF_DataType::TF_HALF: in GetDlDataType()
199 *tf_dtype = TF_DataType::TF_HALF; in TfDataTypeFormDlDataType()
|
| /external/tensorflow/tensorflow/python/lib/core/ |
| D | ndarray_tensor_bridge.cc | 119 case TF_HALF: in TF_DataType_to_PyArray_TYPE()
|
| D | ndarray_tensor.cc | 125 *out_tf_datatype = TF_HALF; in PyArray_TYPE_to_TF_DataType()
|
| /external/tensorflow/tensorflow/go/ |
| D | tensor.go | 67 Half DataType = C.TF_HALF
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_wrapper.cc | 1231 .value("TF_HALF", TF_HALF) in PYBIND11_MODULE()
|