diff options
Diffstat (limited to 'samples/SimpleSample.cpp')
-rw-r--r-- | samples/SimpleSample.cpp | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/samples/SimpleSample.cpp b/samples/SimpleSample.cpp index 3f94b53ca1..01f078bd56 100644 --- a/samples/SimpleSample.cpp +++ b/samples/SimpleSample.cpp @@ -28,7 +28,7 @@ int main() INetworkPtr myNetwork = INetwork::Create(); float weightsData[] = {1.0f}; // Identity - TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32); + TensorInfo weightsInfo(TensorShape({1, 1}), DataType::Float32, 0.0f, 0, true); weightsInfo.SetConstant(); ConstTensor weights(weightsInfo, weightsData); @@ -75,11 +75,12 @@ int main() std::vector<float> inputData{number}; std::vector<float> outputData(1); - - InputTensors inputTensors{{0, armnn::ConstTensor(run->GetInputTensorInfo(networkIdentifier, 0), - inputData.data())}}; + inputTensorInfo = run->GetInputTensorInfo(networkIdentifier, 0); + inputTensorInfo.SetConstant(true); + InputTensors inputTensors{{0, armnn::ConstTensor(inputTensorInfo, + inputData.data())}}; OutputTensors outputTensors{{0, armnn::Tensor(run->GetOutputTensorInfo(networkIdentifier, 0), - outputData.data())}}; + outputData.data())}}; // Execute network run->EnqueueWorkload(networkIdentifier, inputTensors, outputTensors); |