diff options
Diffstat (limited to 'src/armnnTfParser')
-rwxr-xr-x | src/armnnTfParser/TfParser.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp index 2d31842205..3d0c72da4c 100755 --- a/src/armnnTfParser/TfParser.cpp +++ b/src/armnnTfParser/TfParser.cpp @@ -1232,7 +1232,7 @@ ParsedTfOperationPtr TfParser::ParseConv2D(const tensorflow::NodeDef& nodeDef, // Swizzles the content of the tensor's permanent storage into a local storage. std::vector<float> weightTensorSwizzledData(weightTensorInfo.GetNumElements()); armnnUtils::Permute(weightTensorSwizzledInfo.GetShape(), permutationVector, - weightNode->GetStorage(), weightTensorSwizzledData.data()); + weightNode->GetStorage(), weightTensorSwizzledData.data(), sizeof(float)); // Create a weight tensor with the newly swizzled data. ConstTensor weightTensor(weightTensorSwizzledInfo, weightTensorSwizzledData); @@ -1350,7 +1350,7 @@ ParsedTfOperationPtr TfParser::ParseDepthwiseConv2D(const tensorflow::NodeDef& n // Swizzles the content of the tensor's permanent storage into a local storage. std::vector<float> weightTensorSwizzledData(weightTensorInfo.GetNumElements()); armnnUtils::Permute(weightTensorSwizzledInfo.GetShape(), permutationVector, - weightNode->GetStorage(), weightTensorSwizzledData.data()); + weightNode->GetStorage(), weightTensorSwizzledData.data(), sizeof(float)); // Create a weight tensor with the newly swizzled data. ConstTensor weightTensor(weightTensorSwizzledInfo, weightTensorSwizzledData); |