class Tensorflow::OpDefBuilder

Public Class Methods

finalize(pointer) click to toggle source
# File lib/tensorflow/op_def_builder.rb, line 7
def self.finalize(pointer)
  proc do
    FFI::TF_DeleteOpDefinitionBuilder(pointer)
  end
end
new(name) click to toggle source
# File lib/tensorflow/op_def_builder.rb, line 13
def initialize(name)
  @pointer = FFI.TF_NewOpDefinitionBuilder(name)
  ObjectSpace.define_finalizer(self, self.class.finalize(@pointer))
end
unknown_shape_inference_func() click to toggle source
# File lib/tensorflow/op_def_builder.rb, line 3
def self.unknown_shape_inference_func
  @unknown_shape_func ||= FFI.ffi_libraries.first.find_function('TF_ShapeInferenceContextSetUnknownShape')
end

Public Instance Methods

attr(spec) click to toggle source
# File lib/tensorflow/op_def_builder.rb, line 22
def attr(spec)
  FFI.TF_OpDefinitionBuilderAddAttr(self, spec)
  self
end
input(spec) click to toggle source
# File lib/tensorflow/op_def_builder.rb, line 27
def input(spec)
  FFI.TF_OpDefinitionBuilderAddInput(self, spec)
  self
end
output(spec) click to toggle source
# File lib/tensorflow/op_def_builder.rb, line 32
def output(spec)
  FFI.TF_OpDefinitionBuilderAddOutput(self, spec)
  self
end
register() click to toggle source
# File lib/tensorflow/op_def_builder.rb, line 42
def register
  ObjectSpace.undefine_finalizer(self)
  Status.check do |status|
    FFI.TF_RegisterOpDefinition(self, status)
  end
end
shape_inference(func) click to toggle source
# File lib/tensorflow/op_def_builder.rb, line 37
def shape_inference(func)
  FFI.TF_OpDefinitionBuilderSetShapeInferenceFunction(self, func)
  self
end
to_ptr() click to toggle source
# File lib/tensorflow/op_def_builder.rb, line 18
def to_ptr
  @pointer
end