colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rename CopyTensorToData to CopyDataFromTensor to align name of tim::vx::Tensor (#373)

This commit is contained in:
Antkillerfarm 2022-04-24 13:36:51 +08:00 committed by GitHub
parent cccd7860d6
commit dbb3631d4e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
include/tim/vx/platform/native.h
View File

@ -74,7 +74,7 @@ class NativeTensorHandle : public ITensorHandle {
public: public:
NativeTensorHandle(const std::shared_ptr<Tensor>& tensor); NativeTensorHandle(const std::shared_ptr<Tensor>& tensor);
bool CopyDataToTensor(const void* data, uint32_t size_in_bytes) override; bool CopyDataToTensor(const void* data, uint32_t size_in_bytes) override;
bool CopyTensorToData(void* data) override; bool CopyDataFromTensor(void* data) override;
}; };

2
include/tim/vx/platform/platform.h
View File

@ -126,7 +126,7 @@ class ITensorHandle {
public: public:
virtual ~ITensorHandle(){}; virtual ~ITensorHandle(){};
virtual bool CopyDataToTensor(const void* data, uint32_t size_in_bytes) = 0; 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<Tensor> GetTensor() const; virtual std::shared_ptr<Tensor> GetTensor() const;
protected: protected:

6
samples/lenet_multi_device/lenet_multi_device.cc
View File

@ -148,7 +148,7 @@ int main(int argc, char** argv) {
std::vector<uint8_t> input_data0; std::vector<uint8_t> input_data0;
input_data0.resize(28 * 28); 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; std::cout << "Copy intput data fail." << std::endl;
return -1; return -1;
} }
@ -156,7 +156,7 @@ int main(int argc, char** argv) {
std::vector<float> output_data; std::vector<float> output_data;
output_data.resize(1 * 10); 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; std::cout << "Copy output data fail." << std::endl;
return -1; return -1;
} }
@ -164,7 +164,7 @@ int main(int argc, char** argv) {
std::vector<float> output_data1; std::vector<float> output_data1;
output_data1.resize(1 * 10); 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; std::cout << "Copy output data fail." << std::endl;
return -1; return -1;
} }

2
src/tim/vx/platform/native.cc
View File

@ -293,7 +293,7 @@ bool NativeTensorHandle::CopyDataToTensor(const void* data, uint32_t size_in_byt
return tensor_->CopyDataToTensor(data, size_in_bytes); return tensor_->CopyDataToTensor(data, size_in_bytes);
} }
bool NativeTensorHandle::CopyTensorToData(void* data) { bool NativeTensorHandle::CopyDataFromTensor(void* data) {
return tensor_->CopyDataFromTensor(data); return tensor_->CopyDataFromTensor(data);
} }