Browse Source

refactor(mge/imperative): drop ComputingGraphImpl in proxy graph

GitOrigin-RevId: a6e6f2e76b
tags/v1.0.0-rc1
Megvii Engine Team 4 years ago
parent
commit
5c224c71f0
1 changed files with 6 additions and 0 deletions
  1. +6
    -0
      src/core/include/megbrain/graph/static_infer.h

+ 6
- 0
src/core/include/megbrain/graph/static_infer.h View File

@@ -14,6 +14,11 @@
#include "megbrain/tensor.h" #include "megbrain/tensor.h"


namespace mgb { namespace mgb {

namespace imperative {
class ProxyGraph;
} // namespace imperative

namespace cg { namespace cg {


class VarNode; class VarNode;
@@ -50,6 +55,7 @@ namespace static_infer {
const DeviceTensorND *m_value = nullptr; const DeviceTensorND *m_value = nullptr;


friend class StaticInferManagerImpl; friend class StaticInferManagerImpl;
friend class imperative::ProxyGraph;


public: public:
/*! /*!


Loading…
Cancel
Save