Home
last modified time | relevance | path

Searched refs:TF_COMPLEX64 (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.cc82 return TF_COMPLEX64; in GetTensorFlowDataType()
116 case TF_COMPLEX64: in GetTensorFlowLiteType()
Dutil_test.cc119 EXPECT_EQ(TF_COMPLEX64, GetTensorFlowDataType(kTfLiteComplex64)); in TEST()
136 EXPECT_EQ(kTfLiteComplex64, GetTensorFlowLiteType(TF_COMPLEX64)); in TEST()
/external/tensorflow/tensorflow/c/
Dtf_datatype.h55 TF_COMPLEX64 = 8, // Single-precision complex enumerator
Dc_api_test.cc303 EXPECT_EQ(TF_COMPLEX64, static_cast<TF_DataType>(tensorflow::DT_COMPLEX64)); in TEST()
304 EXPECT_EQ(TF_COMPLEX, TF_COMPLEX64); in TEST()
/external/tensorflow/tensorflow/python/lib/core/
Dndarray_tensor_bridge.cc149 case TF_COMPLEX64: in TF_DataType_to_PyArray_TYPE()
Dndarray_tensor.cc153 *out_tf_datatype = TF_COMPLEX64; in PyArray_TYPE_to_TF_DataType()
/external/tensorflow/tensorflow/go/
Dtensor.go54 Complex64 DataType = C.TF_COMPLEX64
401 {reflect.TypeOf(complex(float32(0), float32(0))), C.TF_COMPLEX64},
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc1190 .value("TF_COMPLEX64", TF_COMPLEX64) in PYBIND11_MODULE()