Method: Tensorflow.op_defs

Defined in:
lib/tensorflow-ruby.rb

.op_defs ⇒ Object 



148
149
150
151
152
153
154
155
156
157
 
# File 'lib/tensorflow-ruby.rb', line 148

def self.op_defs
  buffer = FFI.TF_GetAllOpList
  string = buffer[:data].read_string(buffer[:length])
  ops = OpList.decode(string)
  ops.op.each_with_object(Hash.new) do |op_def, hash|
    hash[op_def.name] = op_def
  end
ensure
  FFI.TF_DeleteBuffer(buffer)
end