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