Searched refs:output_tensor_index (Results 1 – 5 of 5) sorted by relevance
78 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in Partition() local79 tensor_epochs_[output_tensor_index] = kEpochNotReady; in Partition()165 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in UpdateNode() local166 tensor_epochs_[output_tensor_index] = current_epoch; in UpdateNode()
195 Index<3> output_tensor_index = FlatToTensorIndex(output_index, output_dims);198 input_tensor_index[0] = output_tensor_index[sp0];199 input_tensor_index[1] = output_tensor_index[sp1];200 input_tensor_index[2] = output_tensor_index[sp2];369 Index<NDIMS> output_tensor_index =373 input_tensor_index[0] = output_tensor_index[0]; // batch376 input_tensor_index[i] = output_tensor_index[i] - padding_left[i - 1];380 input_tensor_index[NDIMS - 1] = output_tensor_index[NDIMS - 1]; // channels398 Index<NDIMS> output_tensor_index =402 input_tensor_index[0] = output_tensor_index[0]; // batch[all …]
201 const int output_tensor_index = interpreter->outputs()[0];203 uint8_t* buffer = interpreter->typed_tensor<uint8_t>(output_tensor_index);
343 const int output_tensor_index = interpreter->outputs()[0];344 TfLiteTensor* output_tensor = interpreter->tensor(output_tensor_index);