Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_kernel.h63 int add_new_ann_tensor_index(int tflite_index) { in add_new_ann_tensor_index() argument
65 if (tflite_index >= current_size) { in add_new_ann_tensor_index()
66 lite_tensor_to_ann_tensor_.resize(tflite_index + 1, -1); in add_new_ann_tensor_index()
69 lite_tensor_to_ann_tensor_[tflite_index] = new_tensor_index; in add_new_ann_tensor_index()
85 void add_type_conversion(int tflite_index, TfLiteType tflite_type) { in add_type_conversion() argument
87 if (tflite_index >= current_size) { in add_type_conversion()
88 index_to_type_conversion_.resize(tflite_index + 1, kTfLiteNoType); in add_type_conversion()
90 index_to_type_conversion_[tflite_index] = tflite_type; in add_type_conversion()