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

Add uid() api for class operation (#668) 

Type: Code Improvement
This commit is contained in:
Zhouheng Zheng 2023-12-18 22:32:21 +08:00 committed by GitHub
parent 11d12f03a8
commit 622c472edf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -48,6 +48,8 @@ class Operation {
uint32_t accumulator_bits = 0);
std::unique_ptr<OpImpl>& impl();
const std::unique_ptr<OpImpl>& impl() const;
const uint32_t& uid() const;
virtual const std::vector<std::shared_ptr<Tensor>> ConstantInputsTensor() const;
protected:

1
src/tim/vx/operation.cc
View File

@ -38,6 +38,7 @@ Operation::~Operation() {}
std::unique_ptr<OpImpl>& Operation::impl() { return impl_; }
const std::unique_ptr<OpImpl>& Operation::impl() const { return impl_; }
const uint32_t& Operation::uid() const { return impl_->node()->uid; }
Operation& Operation::BindInput(const std::shared_ptr<Tensor>& tensor) {
impl_->BindInput(tensor);