Browse Source

Reverted: DefaultGraphStack: Perf-ops

tags/v0.12
Eli Belash 6 years ago
parent
commit
f1bc50ecb7
1 changed files with 12 additions and 12 deletions
  1. +12
    -12
      src/TensorFlowNET.Core/Graphs/DefaultGraphStack.cs

+ 12
- 12
src/TensorFlowNET.Core/Graphs/DefaultGraphStack.cs View File

@@ -27,24 +27,24 @@ namespace Tensorflow
/// </summary> /// </summary>
public class DefaultGraphStack public class DefaultGraphStack
{ {
private readonly List<StackModel> stack = new List<StackModel>();
private readonly List<StackModel> _stack = new List<StackModel>();


public void set_controller(Graph @default) public void set_controller(Graph @default)
{ {
if (!stack.Exists(x => x.Graph == @default))
stack.Add(new StackModel {Graph = @default, IsDefault = true});
if (!_stack.Exists(x => x.Graph == @default))
_stack.Add(new StackModel {Graph = @default, IsDefault = true});


foreach (var s in stack)
foreach (var s in _stack)
s.IsDefault = s.Graph == @default; s.IsDefault = s.Graph == @default;
} }


public Graph get_controller() public Graph get_controller()
{ {
if (stack.Count(x => x.IsDefault) == 0)
stack.Add(new StackModel {Graph = tf.Graph(), IsDefault = true});
for (var i = stack.Count - 1; i >= 0; i--)
if (_stack.Count(x => x.IsDefault) == 0)
_stack.Add(new StackModel {Graph = tf.Graph(), IsDefault = true});
for (var i = _stack.Count - 1; i >= 0; i--)
{ {
var x = stack[i];
var x = _stack[i];
if (x.IsDefault) if (x.IsDefault)
return x.Graph; return x.Graph;
} }
@@ -54,16 +54,16 @@ namespace Tensorflow


public bool remove(Graph g) public bool remove(Graph g)
{ {
if (stack.Count == 0)
if (_stack.Count == 0)
return false; return false;


var sm = stack.Find(model => model.Graph == g);
return sm != null && stack.Remove(sm);
var sm = _stack.Find(model => model.Graph == g);
return sm != null && _stack.Remove(sm);
} }


public void reset() public void reset()
{ {
stack.Clear();
_stack.Clear();
} }


private class StackModel private class StackModel


Loading…
Cancel
Save