Searched refs:kTfLiteComplex64 (Results 1 – 22 of 22) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | complex_support.cc | 37 TF_LITE_ENSURE(context, input->type == kTfLiteComplex64 || in Prepare()
42 if (input->type == kTfLiteComplex64) { in Prepare()
69 case kTfLiteComplex64: { in EvalReal()
101 case kTfLiteComplex64: { in EvalImag()
132 case kTfLiteComplex64: { in EvalAbs()
|
| D | cast.cc | 95 case kTfLiteComplex64: in copyToTensor()
128 case kTfLiteComplex64: in Eval()
|
| D | rfft2d.cc | 205 output->type = kTfLiteComplex64; in Prepare()
437 if (output->type != kTfLiteComplex64) { in Eval()
|
| D | kernel_util.cc | 501 case kTfLiteComplex64: in TfLiteTypeGetSize()
|
| /external/tensorflow/tensorflow/lite/ |
| D | type_to_tflitetype_test.cc | 43 EXPECT_EQ(kTfLiteComplex64, in TEST()
44 typeToTfLiteType<TfLiteTypeToType<kTfLiteComplex64>::Type>()); in TEST()
|
| D | type_to_tflitetype.h | 34 MATCH_TYPE_AND_TFLITE_TYPE(std::complex<float>, kTfLiteComplex64);
|
| D | optional_debug_tools.cc | 70 case kTfLiteComplex64: in TensorTypeName()
|
| D | util.cc | 116 case kTfLiteComplex64: in GetSizeOfType()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 81 case kTfLiteComplex64: in GetTensorFlowDataType()
117 return kTfLiteComplex64; in GetTensorFlowLiteType()
|
| D | util_test.cc | 119 EXPECT_EQ(TF_COMPLEX64, GetTensorFlowDataType(kTfLiteComplex64)); in TEST()
136 EXPECT_EQ(kTfLiteComplex64, GetTensorFlowLiteType(TF_COMPLEX64)); in TEST()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | numpy.cc | 61 case kTfLiteComplex64: in TfLiteTypeToPyArrayType()
104 return kTfLiteComplex64; in TfLiteTypeFromPyType()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_types.h | 70 kTfLiteComplex64 = 8, enumerator
|
| D | common_test.cc | 92 EXPECT_EQ(type_name(kTfLiteComplex64), "COMPLEX64"); in TEST()
|
| D | common.c | 214 case kTfLiteComplex64: in TfLiteTypeGetName()
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | enum_mapping.h | 87 case kTfLiteComplex64: in TfLiteTypeToSchemaType()
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | memory_helpers.cc | 84 case kTfLiteComplex64: in TfLiteTypeSizeOf()
|
| D | memory_helpers_test.cc | 164 tflite::TfLiteTypeSizeOf(kTfLiteComplex64, &size)); in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | tflite_driver.cc | 352 case kTfLiteComplex64: in Check()
541 case kTfLiteComplex64: { in SetInput()
624 case kTfLiteComplex64: in SetExpectation()
|
| /external/tensorflow/tensorflow/lite/objc/sources/ |
| D | TFLInterpreter.mm | 422 case kTfLiteComplex64:
428 // kTfLiteString, kTfLiteUInt64, kTfLiteComplex64, kTfLiteComplex128,
|
| /external/tensorflow/tensorflow/lite/tools/evaluation/stages/ |
| D | topk_accuracy_eval_stage_test.cc | 144 TfLiteType model_output_type = kTfLiteComplex64; in TEST()
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper.cc | 92 case kTfLiteComplex64: in TfLiteTypeToSchemaType()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 876 *type = kTfLiteComplex64; in ConvertTensorType()
|