module TensorStream::CLEventHelpers

Public Instance Methods

build_event_wait_list(inputs) click to toggle source
# 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