From dbb3631d4ec03c746468b840e1cac5576249a300 Mon Sep 17 00:00:00 2001 From: Antkillerfarm Date: Sun, 24 Apr 2022 13:36:51 +0800 Subject: [PATCH] rename CopyTensorToData to CopyDataFromTensor to align name of tim::vx::Tensor (#373) --- include/tim/vx/platform/native.h | 2 +- include/tim/vx/platform/platform.h | 2 +- samples/lenet_multi_device/lenet_multi_device.cc | 6 +++--- src/tim/vx/platform/native.cc | 2 +- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/include/tim/vx/platform/native.h b/include/tim/vx/platform/native.h index 46ca111..e254db2 100644 --- a/include/tim/vx/platform/native.h +++ b/include/tim/vx/platform/native.h @@ -74,7 +74,7 @@ class NativeTensorHandle : public ITensorHandle { public: NativeTensorHandle(const std::shared_ptr& tensor); bool CopyDataToTensor(const void* data, uint32_t size_in_bytes) override; - bool CopyTensorToData(void* data) override; + bool CopyDataFromTensor(void* data) override; }; diff --git a/include/tim/vx/platform/platform.h b/include/tim/vx/platform/platform.h index 64c4bbe..6699427 100644 --- a/include/tim/vx/platform/platform.h +++ b/include/tim/vx/platform/platform.h @@ -126,7 +126,7 @@ class ITensorHandle { public: virtual ~ITensorHandle(){}; virtual bool CopyDataToTensor(const void* data, uint32_t size_in_bytes) = 0; - virtual bool CopyTensorToData(void* data) = 0; + virtual bool CopyDataFromTensor(void* data) = 0; virtual std::shared_ptr GetTensor() const; protected: diff --git a/samples/lenet_multi_device/lenet_multi_device.cc b/samples/lenet_multi_device/lenet_multi_device.cc index d975175..d05dca8 100644 --- a/samples/lenet_multi_device/lenet_multi_device.cc +++ b/samples/lenet_multi_device/lenet_multi_device.cc @@ -148,7 +148,7 @@ int main(int argc, char** argv) { std::vector input_data0; input_data0.resize(28 * 28); - if (!input_handle0->CopyTensorToData(input_data0.data())) { + if (!input_handle0->CopyDataFromTensor(input_data0.data())) { std::cout << "Copy intput data fail." << std::endl; return -1; } @@ -156,7 +156,7 @@ int main(int argc, char** argv) { std::vector output_data; output_data.resize(1 * 10); - if (!output_handle0->CopyTensorToData(output_data.data())) { + if (!output_handle0->CopyDataFromTensor(output_data.data())) { std::cout << "Copy output data fail." << std::endl; return -1; } @@ -164,7 +164,7 @@ int main(int argc, char** argv) { std::vector output_data1; output_data1.resize(1 * 10); - if (!output_handle1->CopyTensorToData(output_data1.data())) { + if (!output_handle1->CopyDataFromTensor(output_data1.data())) { std::cout << "Copy output data fail." << std::endl; return -1; } diff --git a/src/tim/vx/platform/native.cc b/src/tim/vx/platform/native.cc index 2f8c645..33f24ed 100644 --- a/src/tim/vx/platform/native.cc +++ b/src/tim/vx/platform/native.cc @@ -293,7 +293,7 @@ bool NativeTensorHandle::CopyDataToTensor(const void* data, uint32_t size_in_byt return tensor_->CopyDataToTensor(data, size_in_bytes); } -bool NativeTensorHandle::CopyTensorToData(void* data) { +bool NativeTensorHandle::CopyDataFromTensor(void* data) { return tensor_->CopyDataFromTensor(data); }