Method: TensorStream::Evaluator::BaseEvaluator#initialize

Defined in:
lib/tensor_stream/evaluator/base_evaluator.rb

#initialize(session, _device, thread_pool: nil, log_intermediates: false) ⇒ BaseEvaluator

Returns a new instance of BaseEvaluator.



33
34
35
36
37
38
 
# File 'lib/tensor_stream/evaluator/base_evaluator.rb', line 33

def initialize(session, _device, thread_pool: nil, log_intermediates: false)
  @session = session
  @log_intermediates = log_intermediates
  @thread_pool = thread_pool || Concurrent::ImmediateExecutor.new
  @context[:compute_history] = [] if log_intermediates
end