Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/
Dtype_to_tflitetype_test.cc30 typeToTfLiteType<TfLiteTypeToType<kTfLiteInt16>::Type>()); in TEST()
32 typeToTfLiteType<TfLiteTypeToType<kTfLiteInt32>::Type>()); in TEST()
34 typeToTfLiteType<TfLiteTypeToType<kTfLiteUInt32>::Type>()); in TEST()
36 typeToTfLiteType<TfLiteTypeToType<kTfLiteFloat32>::Type>()); in TEST()
38 typeToTfLiteType<TfLiteTypeToType<kTfLiteUInt8>::Type>()); in TEST()
40 typeToTfLiteType<TfLiteTypeToType<kTfLiteInt8>::Type>()); in TEST()
42 typeToTfLiteType<TfLiteTypeToType<kTfLiteBool>::Type>()); in TEST()
44 typeToTfLiteType<TfLiteTypeToType<kTfLiteComplex64>::Type>()); in TEST()
46 typeToTfLiteType<TfLiteTypeToType<kTfLiteComplex128>::Type>()); in TEST()
48 typeToTfLiteType<TfLiteTypeToType<kTfLiteString>::Type>()); in TEST()
[all …]
Dportable_type_to_tflitetype.h39 constexpr TfLiteType typeToTfLiteType() { in typeToTfLiteType() function
51 constexpr TfLiteType typeToTfLiteType<CPP_TYPE>() { \
Dinterpreter.h271 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
283 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
/external/tensorflow/tensorflow/lite/micro/
Dmicro_interpreter.h116 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
132 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_input_tensor()
148 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_output_tensor()
Dtest_helpers.h172 result.type = typeToTfLiteType<T>();
/external/tflite-support/tensorflow_lite_support/cc/task/core/
Dtask_utils.h42 if (tensor_ptr->type == typeToTfLiteType<T>()) { in TypedTensor()
56 TfLiteTypeGetName(typeToTfLiteType<T>()), ", got ", in AssertAndReturnTypedTensor()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dbuffer_map_test.cc43 tensor->type = typeToTfLiteType<T>(); in MakeLiteTensor()
59 tensor->type = typeToTfLiteType<string>(); in MakeLiteTensor()
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h522 CHECK_EQ(t->type, typeToTfLiteType<T>()) in PopulateTensor()
524 << TfLiteTypeGetName(typeToTfLiteType<T>()) << ", got " in PopulateTensor()
544 CHECK_EQ(t->type, typeToTfLiteType<T>()) in PopulateTensor()
546 << TfLiteTypeGetName(typeToTfLiteType<T>()) << ", got " in PopulateTensor()
565 << ". Requested " << typeToTfLiteType<T>() << ", got " in PopulateTensor()