Searched refs:kTfLiteComplex64 (Results 1 – 25 of 26) sorted by relevance
12
| /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 | 104 case kTfLiteComplex64: in copyToTensor()
143 case kTfLiteComplex64: in Eval()
|
| D | mul.cc | 169 } else if (output->type == kTfLiteComplex64) { in EvalMul()
299 if (output->type == kTfLiteComplex64 && params->activation) { in Eval()
305 output->type == kTfLiteInt64 || output->type == kTfLiteComplex64) { in Eval()
|
| D | rfft2d.cc | 205 output->type = kTfLiteComplex64; in Prepare()
437 if (output->type != kTfLiteComplex64) { in Eval()
|
| D | irfft2d.cc | 176 if (input->type != kTfLiteComplex64) { in Prepare()
|
| D | kernel_util.cc | 560 case kTfLiteComplex64: in TfLiteTypeGetSize()
|
| /external/tensorflow/tensorflow/lite/ |
| D | type_to_tflitetype_test.cc | 45 EXPECT_EQ(kTfLiteComplex64, in TEST()
46 typeToTfLiteType<TfLiteTypeToType<kTfLiteComplex64>::Type>()); in TEST()
|
| D | type_to_tflitetype.h | 34 MATCH_TYPE_AND_TFLITE_TYPE(std::complex<float>, kTfLiteComplex64);
|
| D | util.cc | 117 case kTfLiteComplex64: in GetSizeOfType()
|
| D | optional_debug_tools.cc | 332 case kTfLiteComplex64: in TensorTypeName()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util.cc | 95 case kTfLiteComplex64: in GetTensorFlowDataType()
131 return kTfLiteComplex64; in GetTensorFlowLiteType()
172 case kTfLiteComplex64: in TfLiteTypeToTfTypeName()
|
| D | util_test.cc | 126 EXPECT_EQ(TF_COMPLEX64, GetTensorFlowDataType(kTfLiteComplex64)); in TEST()
143 EXPECT_EQ(kTfLiteComplex64, GetTensorFlowLiteType(TF_COMPLEX64)); in TEST()
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | utils.cc | 62 case kTfLiteComplex64: { in CreateRandomTensorData()
152 if (type == kTfLiteComplex64 || type == kTfLiteFloat32 || in GetDataRangesForType()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | numpy.cc | 63 case kTfLiteComplex64: in TfLiteTypeToPyArrayType()
106 return kTfLiteComplex64; in TfLiteTypeFromPyType()
|
| /external/tensorflow/tensorflow/lite/objc/sources/ |
| D | TFLCommonUtil.mm | 47 case kTfLiteComplex64:
54 // kTfLiteString, kTfLiteUInt64, kTfLiteComplex64, kTfLiteComplex128,
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api_types.h | 92 kTfLiteComplex64 = 8, enumerator
|
| D | common_test.cc | 96 EXPECT_EQ(type_name(kTfLiteComplex64), "COMPLEX64"); in TEST()
|
| D | common.cc | 266 case kTfLiteComplex64: in TfLiteTypeGetName()
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | enum_mapping.h | 89 case kTfLiteComplex64: in TfLiteTypeToSchemaType()
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | tflite_driver.cc | 371 case kTfLiteComplex64: in Check()
666 case kTfLiteComplex64: { in SetInput()
752 case kTfLiteComplex64: in SetExpectation()
|
| /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/tools/versioning/ |
| D | op_version.cc | 236 if (op_sig.inputs.at(0).type == kTfLiteComplex64) { in GetBuiltinOperatorVersion()
|
| D | op_version_test.cc | 467 fake_op_sig.inputs = CreateOpSignatureTensorSpecs(kTfLiteComplex64); in TEST()
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper.cc | 100 case kTfLiteComplex64: in TfLiteTypeToSchemaType()
|
| /external/tensorflow/tensorflow/lite/core/api/ |
| D | flatbuffer_conversions.cc | 941 *type = kTfLiteComplex64; in ConvertTensorType()
|
12