Method: TensorStream::OpMaker.define_operation

Defined in:
lib/tensor_stream/op_maker.rb

.define_operation(op_code, &block) ⇒ Object 



47
48
49
50
51
52
 
# File 'lib/tensor_stream/op_maker.rb', line 47

def self.define_operation(op_code, &block)
  @ops ||= {}
  op_maker = TensorStream::OpMaker.new(op_code.to_sym)
  block.call(op_maker)
  @ops[op_code.to_sym] = op_maker
end