diff --git a/include/tim/vx/operation.h b/include/tim/vx/operation.h
index 0effef6..fa96377 100644
--- a/include/tim/vx/operation.h
+++ b/include/tim/vx/operation.h
@@ -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:
diff --git a/src/tim/vx/operation.cc b/src/tim/vx/operation.cc
index 2800b21..8c4d13c 100644
--- a/src/tim/vx/operation.cc
+++ b/src/tim/vx/operation.cc
@@ -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);