Exemplo n.º 1
0
        public virtual void Dispose()
        {
            m_Vars?.Dispose();
            m_Ops?.ResetAllocator(false); // clear allocator's memory
            m_InputShapes?.Clear();

            m_Vars        = null;
            m_Ops         = null;
            m_InputShapes = null;
        }
        public void TakeMemorySnapshot(IOps ops, IVars vars, string context, Layer layer)
        {
            var varsWithStatistics = vars as IVarsStatistics;

            if (varsWithStatistics == null)
            {
                return;
            }

            MemorySnapshotsReports.Add(new MemorySnapshotReport(ops, varsWithStatistics, context, layer));
        }
Exemplo n.º 3
0
        public GenericWorker(Model model, IOps ops, IVars vars, bool verbose = false)
        {
            m_Model             = model;
            m_DefaultInputName  = ModelAnalyzer.GetDefaultInputName(model);
            m_DefaultOutputName = ModelAnalyzer.GetDefaultOutputName(model);
            m_Ops           = ops;
            m_Vars          = vars;
            m_ModelCompiler = ops as IModelCompiler;
            m_Verbose       = verbose;

            m_RequestResetAllocator = true;
        }
Exemplo n.º 4
0
        public IVars Create(IVars argParent, IVars argVars)
        {
            if (argParent == null && argVars != null)
            {
                return(argVars);
            }

            if (argVars == null && argParent != null)
            {
                return(argParent);
            }

            var pVars = NewVars();

            pVars.Parent = argParent;
            pVars.Vars   = argVars;

            return(pVars);
        }
Exemplo n.º 5
0
 /// <inheritdoc/>
 public virtual void PrepareModel(Model model, IDictionary<string, TensorShape> inputShapes, IVars vars)
 {
     if (m_Ops is IModelCompiler)
     ((IModelCompiler)m_Ops).PrepareModel(model, inputShapes, vars);
 }
 public void TakeMemorySnapshot(IVars vars, string context, Layer layer)
 {
     MemorySnapshotsReport.TakeMemorySnapshot(vars, context, layer);
 }
Exemplo n.º 7
0
 public abstract Task <NodeExpr> Eval(NodeExpr e, CancellationToken t, IVars v = null);
Exemplo n.º 8
0
 public IVars CreateAmbito(IVars argVars) => new Ambito(this, argVars);
Exemplo n.º 9
0
 public Ambito(IVars argVars, IVars argAmbito)
 {
     mVars   = argVars;
     mAmbito = argAmbito;
 }
Exemplo n.º 10
0
        public static async Task <NodeExpr> Eval(NodeExpr e, IAlgebra <N> a, CancellationToken t, IVars v = null)
        {
            var vv    = (v != null) ? a.Session.Vars.CreateAmbito(v) : a.Session.Vars;
            var visit = new EvaluateVisitor <N>(a, vv);
            var r     = await visit.Visit(e, t);

            t.ThrowIfCancellationRequested();

            return(r);
        }
Exemplo n.º 11
0
 public EvaluateVisitor(IAlgebra <N> a, IVars v)
 {
     mAlg  = a;
     mVars = v;
 }
Exemplo n.º 12
0
 public Exprs.Expr Eval(Exprs.Expr e, IVars argVars, CancellationToken argCancelToken) => e;