Home
last modified time | relevance | path

Searched refs:tflite_type (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_kernel.h85 void add_type_conversion(int tflite_index, TfLiteType tflite_type) { in add_type_conversion() argument
90 index_to_type_conversion_[tflite_index] = tflite_type; in add_type_conversion()
/external/tensorflow/tensorflow/lite/kernels/
Dmatrix_diag_test.cc80 EXPECT_THAT(model.GetOutputType(), TypeParam::tflite_type); in TYPED_TEST()
Dtest_util.h945 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteFloat32;
955 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteInt32;
965 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteUInt32;
975 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteInt16;
985 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteInt8;
995 static constexpr TfLiteType tflite_type = TfLiteType::kTfLiteUInt8;
Dmatrix_set_diag_test.cc90 EXPECT_THAT(model.GetOutputType(), TypeParam::tflite_type); in TYPED_TEST()