Home
last modified time | relevance | path

Searched refs:TF_COMPLEX128 (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc84 return TF_COMPLEX128; in GetTensorFlowDataType()
118 case TF_COMPLEX128: in GetTensorFlowLiteType()
Dutil_test.cc120 EXPECT_EQ(TF_COMPLEX128, GetTensorFlowDataType(kTfLiteComplex128)); in TEST()
137 EXPECT_EQ(kTfLiteComplex128, GetTensorFlowLiteType(TF_COMPLEX128)); in TEST()
/external/tensorflow/tensorflow/c/
Dtf_datatype.h66 TF_COMPLEX128 = 18, // Double-precision complex enumerator
Dc_api_test.cc314 EXPECT_EQ(TF_COMPLEX128, static_cast<TF_DataType>(tensorflow::DT_COMPLEX128)); in TEST()
2150 TF_SetAttrType(desc, "v", TF_COMPLEX128); in TEST_F()
2159 EXPECT_EQ(TF_COMPLEX128, value); in TEST_F()
2163 const TF_DataType list[] = {TF_FLOAT, TF_DOUBLE, TF_HALF, TF_COMPLEX128}; in TEST_F()
Dkernels_test.cc471 const TF_DataType list[] = {TF_FLOAT, TF_DOUBLE, TF_HALF, TF_COMPLEX128}; in TEST_F()
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor_bridge.cc152 case TF_COMPLEX128: in TF_DataType_to_PyArray_TYPE()
Dndarray_tensor.cc156 *out_tf_datatype = TF_COMPLEX128; in PyArray_TYPE_to_TF_DataType()
/external/tensorflow/tensorflow/go/
Dtensor.go66 Complex128 DataType = C.TF_COMPLEX128
406 {reflect.TypeOf(complex(float64(0), float64(0))), C.TF_COMPLEX128},
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc1201 .value("TF_COMPLEX128", TF_COMPLEX128) in PYBIND11_MODULE()