Lines Matching refs:tensorData

14 void PermuteTensorNchwToNhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData)  in PermuteTensorNchwToNhwc()  argument
20 std::vector<T> tmp(tensorData.size()); in PermuteTensorNchwToNhwc()
21 … armnnUtils::Permute(tensorInfo.GetShape(), nchwToNhwc, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNchwToNhwc()
22 tensorData = tmp; in PermuteTensorNchwToNhwc()
26 void PermuteTensorNhwcToNchw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNhwcToNchw() argument
32 std::vector<T> tmp(tensorData.size()); in PermuteTensorNhwcToNchw()
33 … armnnUtils::Permute(tensorInfo.GetShape(), nhwcToNchw, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNhwcToNchw()
35 tensorData = tmp; in PermuteTensorNhwcToNchw()
39 void PermuteTensorNdhwcToNcdhw(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNdhwcToNcdhw() argument
45 std::vector<T> tmp(tensorData.size()); in PermuteTensorNdhwcToNcdhw()
46 …armnnUtils::Permute(tensorInfo.GetShape(), ndhwcToNcdhw, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNdhwcToNcdhw()
47 tensorData = tmp; in PermuteTensorNdhwcToNcdhw()
51 void PermuteTensorNcdhwToNdhwc(armnn::TensorInfo& tensorInfo, std::vector<T>& tensorData) in PermuteTensorNcdhwToNdhwc() argument
57 std::vector<T> tmp(tensorData.size()); in PermuteTensorNcdhwToNdhwc()
58 …armnnUtils::Permute(tensorInfo.GetShape(), ncdhwToNdhwc, tensorData.data(), tmp.data(), sizeof(T)); in PermuteTensorNcdhwToNdhwc()
59 tensorData = tmp; in PermuteTensorNcdhwToNdhwc()