module TensorStream::Evaluator

Evaluator base module

Public Class Methods

clear_storages(graph) click to toggle source
# File lib/tensor_stream/evaluator/base_evaluator.rb, line 240
def self.clear_storages(graph)
  @storage_managers.values.each { |manager| manager.clear_variables(graph) }
end
default_evaluators() click to toggle source
# File lib/tensor_stream/evaluator/base_evaluator.rb, line 236
def self.default_evaluators
  evaluators.values.sort { |v| v[:index] }.reverse.map { |v| v[:class] }
end
evaluators() click to toggle source
# File lib/tensor_stream/evaluator/base_evaluator.rb, line 225
def self.evaluators
  @evaluators ||= {}
end
read_variable(graph, name) click to toggle source
# File lib/tensor_stream/evaluator/base_evaluator.rb, line 244
def self.read_variable(graph, name)
  @storage_managers.values.each do |manager|
    return manager.read_value(graph, name) if manager.exists?(graph, name)
  end

  nil
end
register_evaluator(klass, name, index = 0) click to toggle source
# File lib/tensor_stream/evaluator/base_evaluator.rb, line 229
def self.register_evaluator(klass, name, index = 0)
  @evaluators ||= {}
  @storage_managers ||= {}
  @evaluators[name] = {name: name, class: klass, index: index}
  @storage_managers[klass] = klass.get_storage_manager
end