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"

namespace mgb {

namespace imperative {
class ProxyGraph;
} // namespace imperative

namespace cg {

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

friend class StaticInferManagerImpl;
friend class imperative::ProxyGraph;

public:
/*!


Loading…
Cancel
Save