Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_plugin.h43 int tflite_index);
59 int tflite_index, TfLiteType tflite_type);
Dnnapi_delegate.cc5406 int tflite_index) { in AddNewNnTensorIndex() argument
5411 if (tflite_index >= current_size) { in AddNewNnTensorIndex()
5412 mapping_context->lite_tensor_to_ann_tensor_.resize(tflite_index + 1, -1); in AddNewNnTensorIndex()
5415 mapping_context->lite_tensor_to_ann_tensor_[tflite_index] = in AddNewNnTensorIndex()
5433 int tflite_index, TfLiteType tflite_type) { in AddTypeConversion() argument
5438 if (tflite_index >= current_size) { in AddTypeConversion()
5439 mapping_context->index_to_type_conversion_.resize(tflite_index + 1, in AddTypeConversion()
5442 mapping_context->index_to_type_conversion_[tflite_index] = tflite_type; in AddTypeConversion()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc353 const int tflite_index = outputs_.TfLiteIndex(i); in MaybePersistTensorflowOutputs() local
354 TfLiteTensor* tensor = &context->tensors[tflite_index]; in MaybePersistTensorflowOutputs()
355 if (!ShouldPersistTensorflowTensor(context, shared_info, tflite_index, in MaybePersistTensorflowOutputs()
358 tflite_index) != kTfLiteOk) { in MaybePersistTensorflowOutputs()