diff --git a/mediapipe/framework/api2/builder.h b/mediapipe/framework/api2/builder.h index 11bcd21c6..111451ce9 100644 --- a/mediapipe/framework/api2/builder.h +++ b/mediapipe/framework/api2/builder.h @@ -120,6 +120,9 @@ using AllowCast = std::integral_constant && } // namespace internal_builder +template +class SourceImpl; + // These classes wrap references to the underlying source/destination // endpoints, adding type information and the user-visible API. template @@ -137,10 +140,14 @@ class DestinationImpl { return DestinationImpl(&base_); } + private: DestinationBase& base_; + + template + friend class SourceImpl; }; -template +template class SourceImpl { public: using Base = SourceBase;