tensorflow-ruby 0.2.0

data/lib/tensorflow.rb

@@ -0,0 +1 @@
+require_relative './tensorflow-ruby'

data/lib/tensorflow/batchable_type_spec.rb

@@ -0,0 +1,4 @@
+module Tensorflow
+  class BatchableTypeSpec < TypeSpec
+  end
+end

data/lib/tensorflow/core/framework/allocation_description_pb.rb

@@ -0,0 +1,21 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/allocation_description.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/allocation_description.proto", :syntax => :proto3) do
+    add_message "tensorflow.AllocationDescription" do
+      optional :requested_bytes, :int64, 1
+      optional :allocated_bytes, :int64, 2
+      optional :allocator_name, :string, 3
+      optional :allocation_id, :int64, 4
+      optional :has_single_reference, :bool, 5
+      optional :ptr, :uint64, 6
+    end
+  end
+end
+
+module Tensorflow
+  AllocationDescription = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.AllocationDescription").msgclass
+end

data/lib/tensorflow/core/framework/api_def_pb.rb

@@ -0,0 +1,59 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/api_def.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/attr_value_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/api_def.proto", :syntax => :proto3) do
+    add_message "tensorflow.ApiDef" do
+      optional :graph_op_name, :string, 1
+      optional :deprecation_message, :string, 12
+      optional :deprecation_version, :int32, 13
+      optional :visibility, :enum, 2, "tensorflow.ApiDef.Visibility"
+      repeated :endpoint, :message, 3, "tensorflow.ApiDef.Endpoint"
+      repeated :in_arg, :message, 4, "tensorflow.ApiDef.Arg"
+      repeated :out_arg, :message, 5, "tensorflow.ApiDef.Arg"
+      repeated :arg_order, :string, 11
+      repeated :attr, :message, 6, "tensorflow.ApiDef.Attr"
+      optional :summary, :string, 7
+      optional :description, :string, 8
+      optional :description_prefix, :string, 9
+      optional :description_suffix, :string, 10
+    end
+    add_message "tensorflow.ApiDef.Endpoint" do
+      optional :name, :string, 1
+      optional :deprecated, :bool, 3
+      optional :deprecation_version, :int32, 4
+    end
+    add_message "tensorflow.ApiDef.Arg" do
+      optional :name, :string, 1
+      optional :rename_to, :string, 2
+      optional :description, :string, 3
+    end
+    add_message "tensorflow.ApiDef.Attr" do
+      optional :name, :string, 1
+      optional :rename_to, :string, 2
+      optional :default_value, :message, 3, "tensorflow.AttrValue"
+      optional :description, :string, 4
+    end
+    add_enum "tensorflow.ApiDef.Visibility" do
+      value :DEFAULT_VISIBILITY, 0
+      value :VISIBLE, 1
+      value :SKIP, 2
+      value :HIDDEN, 3
+    end
+    add_message "tensorflow.ApiDefs" do
+      repeated :op, :message, 1, "tensorflow.ApiDef"
+    end
+  end
+end
+
+module Tensorflow
+  ApiDef = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.ApiDef").msgclass
+  ApiDef::Endpoint = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.ApiDef.Endpoint").msgclass
+  ApiDef::Arg = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.ApiDef.Arg").msgclass
+  ApiDef::Attr = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.ApiDef.Attr").msgclass
+  ApiDef::Visibility = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.ApiDef.Visibility").enummodule
+  ApiDefs = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.ApiDefs").msgclass
+end

data/lib/tensorflow/core/framework/attr_value_pb.rb

@@ -0,0 +1,46 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/attr_value.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/tensor_pb'
+require 'tensorflow/core/framework/tensor_shape_pb'
+require 'tensorflow/core/framework/types_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/attr_value.proto", :syntax => :proto3) do
+    add_message "tensorflow.AttrValue" do
+      oneof :value do
+        optional :s, :bytes, 2
+        optional :i, :int64, 3
+        optional :f, :float, 4
+        optional :b, :bool, 5
+        optional :type, :enum, 6, "tensorflow.DataType"
+        optional :shape, :message, 7, "tensorflow.TensorShapeProto"
+        optional :tensor, :message, 8, "tensorflow.TensorProto"
+        optional :list, :message, 1, "tensorflow.AttrValue.ListValue"
+        optional :func, :message, 10, "tensorflow.NameAttrList"
+        optional :placeholder, :string, 9
+      end
+    end
+    add_message "tensorflow.AttrValue.ListValue" do
+      repeated :s, :bytes, 2
+      repeated :i, :int64, 3
+      repeated :f, :float, 4
+      repeated :b, :bool, 5
+      repeated :type, :enum, 6, "tensorflow.DataType"
+      repeated :shape, :message, 7, "tensorflow.TensorShapeProto"
+      repeated :tensor, :message, 8, "tensorflow.TensorProto"
+      repeated :func, :message, 9, "tensorflow.NameAttrList"
+    end
+    add_message "tensorflow.NameAttrList" do
+      optional :name, :string, 1
+      map :attr, :string, :message, 2, "tensorflow.AttrValue"
+    end
+  end
+end
+
+module Tensorflow
+  AttrValue = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.AttrValue").msgclass
+  AttrValue::ListValue = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.AttrValue.ListValue").msgclass
+  NameAttrList = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.NameAttrList").msgclass
+end

data/lib/tensorflow/core/framework/cost_graph_pb.rb

@@ -0,0 +1,49 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/cost_graph.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/tensor_shape_pb'
+require 'tensorflow/core/framework/types_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/cost_graph.proto", :syntax => :proto3) do
+    add_message "tensorflow.CostGraphDef" do
+      repeated :node, :message, 1, "tensorflow.CostGraphDef.Node"
+    end
+    add_message "tensorflow.CostGraphDef.Node" do
+      optional :name, :string, 1
+      optional :device, :string, 2
+      optional :id, :int32, 3
+      repeated :input_info, :message, 4, "tensorflow.CostGraphDef.Node.InputInfo"
+      repeated :output_info, :message, 5, "tensorflow.CostGraphDef.Node.OutputInfo"
+      optional :temporary_memory_size, :int64, 6
+      optional :persistent_memory_size, :int64, 12
+      optional :host_temp_memory_size, :int64, 10
+      optional :device_temp_memory_size, :int64, 11
+      optional :device_persistent_memory_size, :int64, 16
+      optional :compute_cost, :int64, 9
+      optional :compute_time, :int64, 14
+      optional :memory_time, :int64, 15
+      optional :is_final, :bool, 7
+      repeated :control_input, :int32, 8
+      optional :inaccurate, :bool, 17
+    end
+    add_message "tensorflow.CostGraphDef.Node.InputInfo" do
+      optional :preceding_node, :int32, 1
+      optional :preceding_port, :int32, 2
+    end
+    add_message "tensorflow.CostGraphDef.Node.OutputInfo" do
+      optional :size, :int64, 1
+      optional :alias_input_port, :int64, 2
+      optional :shape, :message, 3, "tensorflow.TensorShapeProto"
+      optional :dtype, :enum, 4, "tensorflow.DataType"
+    end
+  end
+end
+
+module Tensorflow
+  CostGraphDef = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CostGraphDef").msgclass
+  CostGraphDef::Node = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CostGraphDef.Node").msgclass
+  CostGraphDef::Node::InputInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CostGraphDef.Node.InputInfo").msgclass
+  CostGraphDef::Node::OutputInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CostGraphDef.Node.OutputInfo").msgclass
+end

data/lib/tensorflow/core/framework/device_attributes_pb.rb

@@ -0,0 +1,37 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/device_attributes.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/device_attributes.proto", :syntax => :proto3) do
+    add_message "tensorflow.InterconnectLink" do
+      optional :device_id, :int32, 1
+      optional :type, :string, 2
+      optional :strength, :int32, 3
+    end
+    add_message "tensorflow.LocalLinks" do
+      repeated :link, :message, 1, "tensorflow.InterconnectLink"
+    end
+    add_message "tensorflow.DeviceLocality" do
+      optional :bus_id, :int32, 1
+      optional :numa_node, :int32, 2
+      optional :links, :message, 3, "tensorflow.LocalLinks"
+    end
+    add_message "tensorflow.DeviceAttributes" do
+      optional :name, :string, 1
+      optional :device_type, :string, 2
+      optional :memory_limit, :int64, 4
+      optional :locality, :message, 5, "tensorflow.DeviceLocality"
+      optional :incarnation, :fixed64, 6
+      optional :physical_device_desc, :string, 7
+    end
+  end
+end
+
+module Tensorflow
+  InterconnectLink = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.InterconnectLink").msgclass
+  LocalLinks = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.LocalLinks").msgclass
+  DeviceLocality = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.DeviceLocality").msgclass
+  DeviceAttributes = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.DeviceAttributes").msgclass
+end

data/lib/tensorflow/core/framework/function_pb.rb

@@ -0,0 +1,38 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/function.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/attr_value_pb'
+require 'tensorflow/core/framework/node_def_pb'
+require 'tensorflow/core/framework/op_def_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/function.proto", :syntax => :proto3) do
+    add_message "tensorflow.FunctionDefLibrary" do
+      repeated :function, :message, 1, "tensorflow.FunctionDef"
+      repeated :gradient, :message, 2, "tensorflow.GradientDef"
+    end
+    add_message "tensorflow.FunctionDef" do
+      optional :signature, :message, 1, "tensorflow.OpDef"
+      map :attr, :string, :message, 5, "tensorflow.AttrValue"
+      map :arg_attr, :uint32, :message, 7, "tensorflow.FunctionDef.ArgAttrs"
+      repeated :node_def, :message, 3, "tensorflow.NodeDef"
+      map :ret, :string, :string, 4
+      map :control_ret, :string, :string, 6
+    end
+    add_message "tensorflow.FunctionDef.ArgAttrs" do
+      map :attr, :string, :message, 1, "tensorflow.AttrValue"
+    end
+    add_message "tensorflow.GradientDef" do
+      optional :function_name, :string, 1
+      optional :gradient_func, :string, 2
+    end
+  end
+end
+
+module Tensorflow
+  FunctionDefLibrary = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.FunctionDefLibrary").msgclass
+  FunctionDef = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.FunctionDef").msgclass
+  FunctionDef::ArgAttrs = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.FunctionDef.ArgAttrs").msgclass
+  GradientDef = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GradientDef").msgclass
+end

data/lib/tensorflow/core/framework/graph_pb.rb

@@ -0,0 +1,22 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/graph.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/node_def_pb'
+require 'tensorflow/core/framework/function_pb'
+require 'tensorflow/core/framework/versions_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/graph.proto", :syntax => :proto3) do
+    add_message "tensorflow.GraphDef" do
+      repeated :node, :message, 1, "tensorflow.NodeDef"
+      optional :versions, :message, 4, "tensorflow.VersionDef"
+      optional :version, :int32, 3
+      optional :library, :message, 2, "tensorflow.FunctionDefLibrary"
+    end
+  end
+end
+
+module Tensorflow
+  GraphDef = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphDef").msgclass
+end

data/lib/tensorflow/core/framework/graph_transfer_info_pb.rb

@@ -0,0 +1,73 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/graph_transfer_info.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/types_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/graph_transfer_info.proto", :syntax => :proto3) do
+    add_message "tensorflow.GraphTransferNodeInput" do
+      optional :node_id, :int32, 1
+      optional :output_port, :int32, 2
+    end
+    add_message "tensorflow.GraphTransferNodeInfo" do
+      optional :name, :string, 1
+      optional :node_id, :int32, 2
+      optional :type_name, :string, 3
+      optional :soc_op_id, :int32, 4
+      optional :padding_id, :int32, 5
+      optional :input_count, :int32, 6
+      optional :output_count, :int32, 7
+    end
+    add_message "tensorflow.GraphTransferConstNodeInfo" do
+      optional :name, :string, 1
+      optional :node_id, :int32, 2
+      repeated :shape, :int64, 3
+      optional :data, :bytes, 4
+      optional :dtype, :enum, 5, "tensorflow.DataType"
+    end
+    add_message "tensorflow.GraphTransferNodeInputInfo" do
+      optional :node_id, :int32, 1
+      repeated :node_input, :message, 2, "tensorflow.GraphTransferNodeInput"
+    end
+    add_message "tensorflow.GraphTransferNodeOutputInfo" do
+      optional :node_id, :int32, 1
+      repeated :max_byte_size, :int32, 2
+    end
+    add_message "tensorflow.GraphTransferGraphInputNodeInfo" do
+      optional :name, :string, 1
+      repeated :shape, :int64, 2
+      optional :dtype, :enum, 3, "tensorflow.DataType"
+    end
+    add_message "tensorflow.GraphTransferGraphOutputNodeInfo" do
+      optional :name, :string, 1
+      repeated :shape, :int64, 2
+      optional :dtype, :enum, 3, "tensorflow.DataType"
+    end
+    add_message "tensorflow.GraphTransferInfo" do
+      repeated :node_info, :message, 1, "tensorflow.GraphTransferNodeInfo"
+      repeated :const_node_info, :message, 2, "tensorflow.GraphTransferConstNodeInfo"
+      repeated :node_input_info, :message, 3, "tensorflow.GraphTransferNodeInputInfo"
+      repeated :node_output_info, :message, 4, "tensorflow.GraphTransferNodeOutputInfo"
+      repeated :graph_input_node_info, :message, 5, "tensorflow.GraphTransferGraphInputNodeInfo"
+      repeated :graph_output_node_info, :message, 6, "tensorflow.GraphTransferGraphOutputNodeInfo"
+      optional :destination, :enum, 7, "tensorflow.GraphTransferInfo.Destination"
+    end
+    add_enum "tensorflow.GraphTransferInfo.Destination" do
+      value :NOP, 0
+      value :HEXAGON, 1
+    end
+  end
+end
+
+module Tensorflow
+  GraphTransferNodeInput = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphTransferNodeInput").msgclass
+  GraphTransferNodeInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphTransferNodeInfo").msgclass
+  GraphTransferConstNodeInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphTransferConstNodeInfo").msgclass
+  GraphTransferNodeInputInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphTransferNodeInputInfo").msgclass
+  GraphTransferNodeOutputInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphTransferNodeOutputInfo").msgclass
+  GraphTransferGraphInputNodeInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphTransferGraphInputNodeInfo").msgclass
+  GraphTransferGraphOutputNodeInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphTransferGraphOutputNodeInfo").msgclass
+  GraphTransferInfo = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphTransferInfo").msgclass
+  GraphTransferInfo::Destination = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GraphTransferInfo.Destination").enummodule
+end

data/lib/tensorflow/core/framework/kernel_def_pb.rb

@@ -0,0 +1,31 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/kernel_def.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/attr_value_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/kernel_def.proto", :syntax => :proto3) do
+    add_message "tensorflow.KernelDef" do
+      optional :op, :string, 1
+      optional :device_type, :string, 2
+      repeated :constraint, :message, 3, "tensorflow.KernelDef.AttrConstraint"
+      repeated :host_memory_arg, :string, 4
+      optional :label, :string, 5
+      optional :priority, :int32, 6
+    end
+    add_message "tensorflow.KernelDef.AttrConstraint" do
+      optional :name, :string, 1
+      optional :allowed_values, :message, 2, "tensorflow.AttrValue"
+    end
+    add_message "tensorflow.KernelList" do
+      repeated :kernel, :message, 1, "tensorflow.KernelDef"
+    end
+  end
+end
+
+module Tensorflow
+  KernelDef = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.KernelDef").msgclass
+  KernelDef::AttrConstraint = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.KernelDef.AttrConstraint").msgclass
+  KernelList = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.KernelList").msgclass
+end

data/lib/tensorflow/core/framework/log_memory_pb.rb

@@ -0,0 +1,53 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/framework/log_memory.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/tensor_description_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/framework/log_memory.proto", :syntax => :proto3) do
+    add_message "tensorflow.MemoryLogStep" do
+      optional :step_id, :int64, 1
+      optional :handle, :string, 2
+    end
+    add_message "tensorflow.MemoryLogTensorAllocation" do
+      optional :step_id, :int64, 1
+      optional :kernel_name, :string, 2
+      optional :tensor, :message, 3, "tensorflow.TensorDescription"
+    end
+    add_message "tensorflow.MemoryLogTensorDeallocation" do
+      optional :allocation_id, :int64, 1
+      optional :allocator_name, :string, 2
+    end
+    add_message "tensorflow.MemoryLogTensorOutput" do
+      optional :step_id, :int64, 1
+      optional :kernel_name, :string, 2
+      optional :index, :int32, 3
+      optional :tensor, :message, 4, "tensorflow.TensorDescription"
+    end
+    add_message "tensorflow.MemoryLogRawAllocation" do
+      optional :step_id, :int64, 1
+      optional :operation, :string, 2
+      optional :num_bytes, :int64, 3
+      optional :ptr, :uint64, 4
+      optional :allocation_id, :int64, 5
+      optional :allocator_name, :string, 6
+    end
+    add_message "tensorflow.MemoryLogRawDeallocation" do
+      optional :step_id, :int64, 1
+      optional :operation, :string, 2
+      optional :allocation_id, :int64, 3
+      optional :allocator_name, :string, 4
+      optional :deferred, :bool, 5
+    end
+  end
+end
+
+module Tensorflow
+  MemoryLogStep = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MemoryLogStep").msgclass
+  MemoryLogTensorAllocation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MemoryLogTensorAllocation").msgclass
+  MemoryLogTensorDeallocation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MemoryLogTensorDeallocation").msgclass
+  MemoryLogTensorOutput = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MemoryLogTensorOutput").msgclass
+  MemoryLogRawAllocation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MemoryLogRawAllocation").msgclass
+  MemoryLogRawDeallocation = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MemoryLogRawDeallocation").msgclass
+end