lib/tensorflow/ffi.rb in tensorflow-0.1.1 vs lib/tensorflow/ffi.rb in tensorflow-0.1.2
- old
+ new
@@ -33,10 +33,14 @@
# https://github.com/tensorflow/tensorflow/blob/master/tensorflow/c/tf_tensor.h
attach_function :TF_NewTensor, %i[int pointer int pointer size_t pointer pointer], :pointer
attach_function :TF_DeleteTensor, %i[pointer], :void
attach_function :TF_TensorData, %i[pointer], :pointer
+ attach_function :TF_TensorByteSize, %i[pointer], :size_t
+ attach_function :TF_StringEncode, %i[pointer size_t pointer size_t pointer], :size_t
+ attach_function :TF_StringDecode, %i[pointer size_t pointer pointer pointer], :size_t
+ attach_function :TF_StringEncodedSize, %i[size_t], :size_t
# https://github.com/tensorflow/tensorflow/blob/master/tensorflow/c/eager/c_api.h
attach_function :TFE_NewContextOptions, %i[], :pointer
attach_function :TFE_ContextOptionsSetAsync, %i[pointer char], :void
attach_function :TFE_DeleteContextOptions, %i[pointer], :void
@@ -56,12 +60,20 @@
attach_function :TFE_NewOp, %i[pointer string pointer], :pointer
attach_function :TFE_DeleteOp, %i[pointer], :void
attach_function :TFE_OpSetDevice, %i[pointer string pointer], :pointer
attach_function :TFE_OpGetDevice, %i[pointer pointer], :string
attach_function :TFE_OpAddInput, %i[pointer pointer pointer], :void
+ attach_function :TFE_OpAddInputList, %i[pointer pointer int pointer], :void
attach_function :TFE_OpGetAttrType, %i[pointer string pointer pointer], :int
attach_function :TFE_OpSetAttrString, %i[pointer string pointer size_t], :void
+ attach_function :TFE_OpSetAttrInt, %i[pointer string int64_t], :void
+ attach_function :TFE_OpSetAttrFloat, %i[pointer string float], :void
+ attach_function :TFE_OpSetAttrBool, %i[pointer string uint8], :void
attach_function :TFE_OpSetAttrType, %i[pointer string int], :void
attach_function :TFE_OpSetAttrShape, %i[pointer string pointer int pointer], :void
+ attach_function :TFE_OpSetAttrIntList, %i[pointer string pointer int], :void
+ attach_function :TFE_OpSetAttrFloatList, %i[pointer string pointer int], :void
+ attach_function :TFE_OpSetAttrTypeList, %i[pointer string pointer int], :void
+ attach_function :TFE_OpSetAttrShapeList, %i[pointer string pointer pointer int pointer], :void
attach_function :TFE_Execute, %i[pointer pointer pointer pointer], :pointer
end
end