experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Metal SingleOpModel: fixed incorrect indexing in/out tensors' ids.

Browse Source 
PiperOrigin-RevId: 258986398
This commit is contained in:
A. Unique TensorFlower 2019-07-19 09:50:29 -07:00 committed by TensorFlower Gardener
parent d2dec8feff
commit 922346b49f
1 changed files with 1 additions and 2 deletions

3
tensorflow/lite/delegates/gpu/metal/kernels/test_util.mm
View File

@ -56,7 +56,7 @@ SingleOpModel::SingleOpModel(Operation&& operation, const std::vector<TensorRef<
output->tensor = outputs[i];
graph_.SetProducer(node->id, output->id).IgnoreError();
TensorFloat32 tensor;
tensor.id = output->tensor.ref;
tensor.id = output->id;
tensor.shape = output->tensor.shape;
outputs_.emplace_back(std::move(tensor));
}
@ -79,7 +79,6 @@ Status SingleOpModel::Invoke() {
options.accumulator_precision = RuntimeOptions::Precision::FP32;
CompiledModel compiled_model;
RETURN_IF_ERROR(Compile(graph_, options, &compiled_model));
std::string err = "res: ";
CompiledModel optimized_model;
RETURN_IF_ERROR(ValidateOptimizeModel(input_ids, output_ids, compiled_model, &optimized_model));