Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc87 int TfLiteIndex(int i) const { return inputs_[i]; } in TfLiteIndex() function in tflite::flex::OpInputs
162 int TfLiteIndex(int i) const { return outputs_[i]; } in TfLiteIndex() function in tflite::flex::OpOutputs
277 int input_index = inputs_.TfLiteIndex(i); in BuildEagerInputs()
309 buffer_map->SetFromTensorFlow(outputs_.TfLiteIndex(i), *tensor); in PersistEagerOutputs()
430 int output_index = node_data->outputs().TfLiteIndex(i); in Init()
498 ++tensor_ref_count[node_data->inputs().TfLiteIndex(i)]; in Prepare()
507 bool f = (tensor_ref_count[node_data->inputs().TfLiteIndex(i)] == 1); in Prepare()
531 const auto input_tensor_index = node_data->inputs().TfLiteIndex(i); in ValidateOutputTensorShapeConsistency()
572 const auto output_tensor_index = node_data->outputs().TfLiteIndex(i); in ValidateOutputTensorShapeConsistency()