Module: TensorStream::CLEventHelpers

Included in:
Evaluator::OpenclEvaluator, OpenCLBuffer
Defined in:
lib/tensor_stream/opencl/utils.rb

Instance Method Summary collapse

Instance Method Details

#build_event_wait_list(inputs) ⇒ Object 



19
20
21
22
23
24
25
 
# File 'lib/tensor_stream/opencl/utils.rb', line 19

def build_event_wait_list(inputs)
  if inputs.is_a?(Array)
    inputs.flatten.compact.map(&:op).compact.uniq
  else
    inputs.op ? [inputs.op] : []
  end
end