"examples/vscode:/vscode.git/clone" did not exist on "5f150c4cef2140342c66ef44ba8f36b5844a51b1"
Commit 9e03c383 authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Use type alias to replace duplicated code

parent 95f21ea5
...@@ -235,7 +235,8 @@ auto make_ParallelTensorFunctor(F f, Xs... xs) ...@@ -235,7 +235,8 @@ auto make_ParallelTensorFunctor(F f, Xs... xs)
template <typename T> template <typename T>
struct Tensor struct Tensor
{ {
using Data = std::vector<T>; using Descriptor = HostTensorDescriptor;
using Data = std::vector<T>;
template <typename X> template <typename X>
Tensor(std::initializer_list<X> lens) : mDesc(lens), mData(mDesc.GetElementSpaceSize()) Tensor(std::initializer_list<X> lens) : mDesc(lens), mData(mDesc.GetElementSpaceSize())
...@@ -253,7 +254,7 @@ struct Tensor ...@@ -253,7 +254,7 @@ struct Tensor
{ {
} }
Tensor(const HostTensorDescriptor& desc) : mDesc(desc), mData(mDesc.GetElementSpaceSize()) {} Tensor(const Descriptor& desc) : mDesc(desc), mData(mDesc.GetElementSpaceSize()) {}
template <typename OutT> template <typename OutT>
Tensor<OutT> CopyAsType() const Tensor<OutT> CopyAsType() const
...@@ -280,9 +281,9 @@ struct Tensor ...@@ -280,9 +281,9 @@ struct Tensor
{ {
} }
const std::vector<std::size_t>& GetLengths() const { return mDesc.GetLengths(); } decltype(auto) GetLengths() const { return mDesc.GetLengths(); }
const std::vector<std::size_t>& GetStrides() const { return mDesc.GetStrides(); } decltype(auto) GetStrides() const { return mDesc.GetStrides(); }
std::size_t GetNumOfDimension() const { return mDesc.GetNumOfDimension(); } std::size_t GetNumOfDimension() const { return mDesc.GetNumOfDimension(); }
...@@ -441,6 +442,6 @@ struct Tensor ...@@ -441,6 +442,6 @@ struct Tensor
typename Data::const_pointer data() const { return mData.data(); } typename Data::const_pointer data() const { return mData.data(); }
HostTensorDescriptor mDesc; Descriptor mDesc;
Data mData; Data mData;
}; };
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment