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