Searched refs:output_tensor_index (Results 1 – 6 of 6) sorted by relevance
199 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims);202 input_tensor_index[0] = output_tensor_index[sp0];203 input_tensor_index[1] = output_tensor_index[sp1];204 input_tensor_index[2] = output_tensor_index[sp2];235 Index<3> output_tensor_index =238 input_tensor_index[0] = output_tensor_index[sp0];239 input_tensor_index[1] = output_tensor_index[sp1];240 input_tensor_index[2] = output_tensor_index[sp2];250 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims);253 input_tensor_index[0] = output_tensor_index[sp0];[all …]
64 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in Partition() local65 tensor_epochs_[output_tensor_index] = kEpochNotReady; in Partition()153 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in UpdateNode() local154 tensor_epochs_[output_tensor_index] = current_epoch; in UpdateNode()
180 int output_tensor_index = node->outputs->data[i]; in ResizeOutputTensors() local181 TfLiteTensor* output_tensor = &context->tensors[output_tensor_index]; in ResizeOutputTensors()
572 const auto output_tensor_index = node_data->outputs().TfLiteIndex(i); in ValidateOutputTensorShapeConsistency() local573 TfLiteTensor* tfl_tensor = &context->tensors[output_tensor_index]; in ValidateOutputTensorShapeConsistency()585 op_name, i, output_tensor_index, tfl_shape_string.c_str(), in ValidateOutputTensorShapeConsistency()
1122 int output_tensor_index = op->outputs[pair.index()]; in ConvertSubgraph() local1123 auto& tensor = *subgraph.tensors[output_tensor_index]; in ConvertSubgraph()1126 vals_map[output_tensor_index] = stats_op->getResult(0); in ConvertSubgraph()1128 vals_map[output_tensor_index] = pair.value(); in ConvertSubgraph()
344 const int output_tensor_index = interpreter->outputs()[0];345 TfLiteTensor* output_tensor = interpreter->tensor(output_tensor_index);