Lines Matching refs:weightsInfo
1447 armnn::TensorInfo weightsInfo = input.GetInfo(); in ParseConstant() local
1448 … std::unique_ptr<unsigned char[]> permuteBuffer(new unsigned char[weightsInfo.GetNumBytes()]); in ParseConstant()
1449 weightsInfo = armnnUtils::Permuted(weightsInfo, permutationVector); in ParseConstant()
1450 armnnUtils::Permute(weightsInfo.GetShape(), permutationVector, in ParseConstant()
1452 GetDataTypeSize(weightsInfo.GetDataType())); in ParseConstant()
1455 auto weightsShape = weightsInfo.GetShape(); in ParseConstant()
1456 weightsInfo.SetShape({1, in ParseConstant()
1460 weightsInfo.SetConstant(true); in ParseConstant()
1462 armnn::ConstTensor weightsPermuted(weightsInfo, permuteBuffer.get()); in ParseConstant()
1680 armnn::TensorInfo weightsInfo = weights.GetInfo(); in ParseDepthwiseConvolution2d() local
1681 … std::unique_ptr<unsigned char[]> permuteBuffer(new unsigned char[weightsInfo.GetNumBytes()]); in ParseDepthwiseConvolution2d()
1682 weightsInfo = armnnUtils::Permuted(weightsInfo, permutationVector); in ParseDepthwiseConvolution2d()
1683 armnnUtils::Permute(weightsInfo.GetShape(), permutationVector, in ParseDepthwiseConvolution2d()
1685 GetDataTypeSize(weightsInfo.GetDataType())); in ParseDepthwiseConvolution2d()
1688 auto weightsShape = weightsInfo.GetShape(); in ParseDepthwiseConvolution2d()
1689 weightsInfo.SetShape({1, in ParseDepthwiseConvolution2d()
1694 armnn::ConstTensor weightsPermuted(weightsInfo, permuteBuffer.get()); in ParseDepthwiseConvolution2d()