Searched refs:typeToTfLiteType (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | type_to_tflitetype_test.cc | 30 typeToTfLiteType<TfLiteTypeToType<kTfLiteInt16>::Type>()); in TEST()
32 typeToTfLiteType<TfLiteTypeToType<kTfLiteUInt16>::Type>()); in TEST()
34 typeToTfLiteType<TfLiteTypeToType<kTfLiteInt32>::Type>()); in TEST()
36 typeToTfLiteType<TfLiteTypeToType<kTfLiteUInt32>::Type>()); in TEST()
38 typeToTfLiteType<TfLiteTypeToType<kTfLiteFloat32>::Type>()); in TEST()
40 typeToTfLiteType<TfLiteTypeToType<kTfLiteUInt8>::Type>()); in TEST()
42 typeToTfLiteType<TfLiteTypeToType<kTfLiteInt8>::Type>()); in TEST()
44 typeToTfLiteType<TfLiteTypeToType<kTfLiteBool>::Type>()); in TEST()
46 typeToTfLiteType<TfLiteTypeToType<kTfLiteComplex64>::Type>()); in TEST()
48 typeToTfLiteType<TfLiteTypeToType<kTfLiteComplex128>::Type>()); in TEST()
[all …]
|
| D | portable_type_to_tflitetype.h | 39 constexpr TfLiteType typeToTfLiteType() { in typeToTfLiteType() function
51 constexpr TfLiteType typeToTfLiteType<CPP_TYPE>() { \
|
| D | interpreter.h | 279 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
291 if (tensor_ptr->type == typeToTfLiteType<T>()) { in typed_tensor()
|
| /external/tensorflow/tensorflow/lite/kernels/shim/ |
| D | test_util.cc | 45 TFLITE_DCHECK_EQ(tensor->type, ::tflite::typeToTfLiteType<T>()); in TensorValueToString()
53 TFLITE_DCHECK_EQ(tensor->type, ::tflite::typeToTfLiteType<bool>()); in TensorValueToString()
61 TFLITE_DCHECK_EQ(tensor->type, ::tflite::typeToTfLiteType<FloatType>()); in TensorValueToStringFloat()
|
| D | test_util.h | 70 tensor->type = typeToTfLiteType<T>(); in ReallocDynamicTensor()
|
| /external/tflite-support/tensorflow_lite_support/cc/task/core/ |
| D | task_utils.h | 42 if (tensor_ptr->type == typeToTfLiteType<T>()) { in TypedTensor()
56 TfLiteTypeGetName(typeToTfLiteType<T>()), ", got ", in AssertAndReturnTypedTensor()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | buffer_map_test.cc | 51 tensor->type = typeToTfLiteType<T>(); in MakeLiteTensor()
67 tensor->type = typeToTfLiteType<string>(); in MakeLiteTensor()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.h | 566 CHECK_EQ(t->type, typeToTfLiteType<T>()) in PopulateTensor()
568 << TfLiteTypeGetName(typeToTfLiteType<T>()) << ", got " in PopulateTensor()
588 CHECK_EQ(t->type, typeToTfLiteType<T>()) in PopulateTensor()
590 << TfLiteTypeGetName(typeToTfLiteType<T>()) << ", got " in PopulateTensor()
609 << ". Requested " << typeToTfLiteType<T>() << ", got " in PopulateTensor()
|