Lines Matching refs:tensorInfo
14 void PermuteTensorNchwToNhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNchwToNhwc() argument
18 tensorInfo = armnnUtils::Permuted(tensorInfo, nchwToNhwc); in PermuteTensorNchwToNhwc()
21 … armnnUtils::Permute(tensorInfo.GetShape(), nchwToNhwc, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNchwToNhwc()
26 void PermuteTensorNhwcToNchw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNhwcToNchw() argument
30 tensorInfo = armnnUtils::Permuted(tensorInfo, nhwcToNchw); in PermuteTensorNhwcToNchw()
33 … armnnUtils::Permute(tensorInfo.GetShape(), nhwcToNchw, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNhwcToNchw()
39 void PermuteTensorNdhwcToNcdhw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNdhwcToNcdhw() argument
43 tensorInfo = armnnUtils::Permuted(tensorInfo, ndhwcToNcdhw); in PermuteTensorNdhwcToNcdhw()
46 …armnnUtils::Permute(tensorInfo.GetShape(), ndhwcToNcdhw, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNdhwcToNcdhw()
51 void PermuteTensorNcdhwToNdhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNcdhwToNdhwc() argument
55 tensorInfo = armnnUtils::Permuted(tensorInfo, ncdhwToNdhwc); in PermuteTensorNcdhwToNdhwc()
58 …armnnUtils::Permute(tensorInfo.GetShape(), ncdhwToNdhwc, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNcdhwToNdhwc()