tensorflow-ruby 0.2.0

data/lib/tensorflow/core/protobuf/verifier_config_pb.rb

@@ -0,0 +1,23 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/protobuf/verifier_config.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/protobuf/verifier_config.proto", :syntax => :proto3) do
+    add_message "tensorflow.VerifierConfig" do
+      optional :verification_timeout_in_ms, :int64, 1
+      optional :structure_verifier, :enum, 2, "tensorflow.VerifierConfig.Toggle"
+    end
+    add_enum "tensorflow.VerifierConfig.Toggle" do
+      value :DEFAULT, 0
+      value :ON, 1
+      value :OFF, 2
+    end
+  end
+end
+
+module Tensorflow
+  VerifierConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.VerifierConfig").msgclass
+  VerifierConfig::Toggle = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.VerifierConfig.Toggle").enummodule
+end

data/lib/tensorflow/core/protobuf/worker_pb.rb

@@ -0,0 +1,246 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/protobuf/worker.proto
+
+require 'google/protobuf'
+
+require 'google/protobuf/any_pb'
+require 'tensorflow/core/framework/cost_graph_pb'
+require 'tensorflow/core/framework/device_attributes_pb'
+require 'tensorflow/core/framework/graph_pb'
+require 'tensorflow/core/framework/step_stats_pb'
+require 'tensorflow/core/framework/tensor_pb'
+require 'tensorflow/core/framework/tensor_shape_pb'
+require 'tensorflow/core/framework/types_pb'
+require 'tensorflow/core/lib/core/error_codes_pb'
+require 'tensorflow/core/protobuf/config_pb'
+require 'tensorflow/core/protobuf/debug_pb'
+require 'tensorflow/core/protobuf/named_tensor_pb'
+require 'tensorflow/core/protobuf/tensorflow_server_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/protobuf/worker.proto", :syntax => :proto3) do
+    add_message "tensorflow.GetStatusRequest" do
+    end
+    add_message "tensorflow.GetStatusResponse" do
+      repeated :device_attributes, :message, 1, "tensorflow.DeviceAttributes"
+    end
+    add_message "tensorflow.CreateWorkerSessionRequest" do
+      optional :session_handle, :string, 1
+      optional :server_def, :message, 2, "tensorflow.ServerDef"
+      optional :isolate_session_state, :bool, 3
+      repeated :cluster_device_attributes, :message, 4, "tensorflow.DeviceAttributes"
+    end
+    add_message "tensorflow.CreateWorkerSessionResponse" do
+    end
+    add_message "tensorflow.DeleteWorkerSessionRequest" do
+      optional :session_handle, :string, 1
+    end
+    add_message "tensorflow.DeleteWorkerSessionResponse" do
+    end
+    add_message "tensorflow.RegisterGraphRequest" do
+      optional :session_handle, :string, 1
+      optional :create_worker_session_called, :bool, 6
+      optional :graph_def, :message, 2, "tensorflow.GraphDef"
+      optional :has_control_flow, :bool, 3
+      optional :graph_options, :message, 4, "tensorflow.GraphOptions"
+      optional :debug_options, :message, 5, "tensorflow.DebugOptions"
+      optional :collective_graph_key, :int64, 7
+    end
+    add_message "tensorflow.RegisterGraphResponse" do
+      optional :graph_handle, :string, 1
+    end
+    add_message "tensorflow.DeregisterGraphRequest" do
+      optional :session_handle, :string, 2
+      optional :create_worker_session_called, :bool, 3
+      optional :graph_handle, :string, 1
+    end
+    add_message "tensorflow.DeregisterGraphResponse" do
+    end
+    add_message "tensorflow.CleanupAllRequest" do
+      repeated :container, :string, 1
+    end
+    add_message "tensorflow.CleanupAllResponse" do
+    end
+    add_message "tensorflow.ExecutorOpts" do
+      optional :record_costs, :bool, 1
+      optional :record_timeline, :bool, 3
+      optional :record_partition_graphs, :bool, 4
+      optional :report_tensor_allocations_upon_oom, :bool, 5
+    end
+    add_message "tensorflow.RunGraphRequest" do
+      optional :session_handle, :string, 8
+      optional :create_worker_session_called, :bool, 10
+      optional :graph_handle, :string, 1
+      optional :step_id, :int64, 2
+      optional :exec_opts, :message, 5, "tensorflow.ExecutorOpts"
+      repeated :send, :message, 3, "tensorflow.NamedTensorProto"
+      repeated :recv_key, :string, 4
+      optional :is_partial, :bool, 6
+      optional :is_last_partial_run, :bool, 7
+      optional :store_errors_in_response_body, :bool, 9
+      optional :request_id, :int64, 11
+    end
+    add_message "tensorflow.RunGraphResponse" do
+      repeated :recv, :message, 1, "tensorflow.NamedTensorProto"
+      optional :step_stats, :message, 2, "tensorflow.StepStats"
+      optional :cost_graph, :message, 3, "tensorflow.CostGraphDef"
+      repeated :partition_graph, :message, 4, "tensorflow.GraphDef"
+      optional :status_code, :enum, 5, "tensorflow.error.Code"
+      optional :status_error_message, :string, 6
+    end
+    add_message "tensorflow.CleanupGraphRequest" do
+      optional :step_id, :int64, 1
+    end
+    add_message "tensorflow.CleanupGraphResponse" do
+    end
+    add_message "tensorflow.RecvTensorRequest" do
+      optional :step_id, :int64, 1
+      optional :rendezvous_key, :string, 2
+      optional :dma_ok, :bool, 3
+      optional :client_locality, :message, 4, "tensorflow.DeviceLocality"
+      optional :server_locality, :message, 5, "tensorflow.DeviceLocality"
+      optional :transport_options, :message, 6, "google.protobuf.Any"
+      optional :request_id, :int64, 7
+    end
+    add_message "tensorflow.RecvTensorResponse" do
+      optional :tensor, :message, 1, "tensorflow.TensorProto"
+      optional :is_dead, :bool, 2
+      optional :send_start_micros, :int64, 3
+      optional :transport_options, :message, 4, "google.protobuf.Any"
+      optional :require_ack, :bool, 5
+    end
+    add_message "tensorflow.MarkRecvFinishedRequest" do
+      optional :request_id, :int64, 1
+    end
+    add_message "tensorflow.MarkRecvFinishedResponse" do
+    end
+    add_message "tensorflow.LoggingRequest" do
+      optional :enable_rpc_logging, :bool, 1
+      optional :disable_rpc_logging, :bool, 4
+      optional :clear, :bool, 2
+      repeated :fetch_step_id, :int64, 3
+    end
+    add_message "tensorflow.LabeledStepStats" do
+      optional :step_id, :int64, 1
+      optional :step_stats, :message, 2, "tensorflow.StepStats"
+    end
+    add_message "tensorflow.LoggingResponse" do
+      repeated :step, :message, 1, "tensorflow.LabeledStepStats"
+    end
+    add_message "tensorflow.TraceOpts" do
+      optional :duration, :double, 1
+      optional :use_step_profiler, :bool, 2
+      optional :use_kernel_profiler, :bool, 3
+      optional :use_extended_profiler, :bool, 4
+      optional :use_gpu_profiler, :bool, 5
+      optional :use_sample_profiler, :bool, 6
+    end
+    add_message "tensorflow.TracingRequest" do
+      optional :options, :message, 1, "tensorflow.TraceOpts"
+    end
+    add_message "tensorflow.TracingResponse" do
+    end
+    add_message "tensorflow.RecvBufRequest" do
+      optional :step_id, :int64, 1
+      optional :buf_rendezvous_key, :string, 2
+      optional :num_bytes, :int64, 3
+      optional :buf_ptr, :fixed64, 4
+      optional :client_locality, :message, 5, "tensorflow.DeviceLocality"
+      optional :server_locality, :message, 6, "tensorflow.DeviceLocality"
+      optional :transport_options, :message, 7, "google.protobuf.Any"
+      optional :src_device, :string, 8
+      optional :dst_device, :string, 9
+      optional :request_id, :int64, 10
+      optional :src_incarnation, :uint64, 11
+    end
+    add_message "tensorflow.RecvBufResponse" do
+      optional :buf_ptr, :fixed64, 1
+      optional :num_bytes, :int64, 2
+      optional :is_dead, :bool, 3
+      optional :transport_options, :message, 4, "google.protobuf.Any"
+      optional :send_start_micros, :int64, 5
+      optional :require_ack, :bool, 6
+    end
+    add_message "tensorflow.CompleteGroupRequest" do
+      optional :group_key, :int32, 1
+      optional :group_size, :int32, 2
+      optional :device_type, :string, 3
+      repeated :device_name, :string, 4
+      optional :collective_type, :int32, 5
+    end
+    add_message "tensorflow.CompleteGroupResponse" do
+      optional :group_key, :int32, 1
+      optional :group_size, :int32, 2
+      optional :device_type, :string, 3
+      optional :num_tasks, :int32, 4
+      repeated :device_name, :string, 5
+      repeated :task_name, :string, 6
+      optional :communicator_key, :bytes, 7
+    end
+    add_message "tensorflow.CompleteInstanceRequest" do
+      optional :name, :string, 1
+      optional :type, :int32, 2
+      optional :data_type, :enum, 3, "tensorflow.DataType"
+      optional :shape, :message, 4, "tensorflow.TensorShapeProto"
+      optional :group_key, :int32, 5
+      optional :group_size, :int32, 6
+      optional :instance_key, :int32, 7
+      optional :device_type, :string, 8
+      repeated :subdiv_offset, :int32, 9
+      optional :device, :string, 10
+      optional :is_source, :bool, 11
+    end
+    add_message "tensorflow.CompleteInstanceResponse" do
+      optional :instance_key, :int32, 1
+      optional :source_rank, :int32, 2
+    end
+    add_message "tensorflow.GetStepSequenceRequest" do
+      repeated :graph_key, :int64, 1
+    end
+    add_message "tensorflow.StepSequence" do
+      optional :graph_key, :int64, 1
+      optional :next_step_id, :int64, 2
+    end
+    add_message "tensorflow.GetStepSequenceResponse" do
+      repeated :step_sequence, :message, 1, "tensorflow.StepSequence"
+    end
+  end
+end
+
+module Tensorflow
+  GetStatusRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GetStatusRequest").msgclass
+  GetStatusResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GetStatusResponse").msgclass
+  CreateWorkerSessionRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CreateWorkerSessionRequest").msgclass
+  CreateWorkerSessionResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CreateWorkerSessionResponse").msgclass
+  DeleteWorkerSessionRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.DeleteWorkerSessionRequest").msgclass
+  DeleteWorkerSessionResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.DeleteWorkerSessionResponse").msgclass
+  RegisterGraphRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.RegisterGraphRequest").msgclass
+  RegisterGraphResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.RegisterGraphResponse").msgclass
+  DeregisterGraphRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.DeregisterGraphRequest").msgclass
+  DeregisterGraphResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.DeregisterGraphResponse").msgclass
+  CleanupAllRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CleanupAllRequest").msgclass
+  CleanupAllResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CleanupAllResponse").msgclass
+  ExecutorOpts = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.ExecutorOpts").msgclass
+  RunGraphRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.RunGraphRequest").msgclass
+  RunGraphResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.RunGraphResponse").msgclass
+  CleanupGraphRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CleanupGraphRequest").msgclass
+  CleanupGraphResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CleanupGraphResponse").msgclass
+  RecvTensorRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.RecvTensorRequest").msgclass
+  RecvTensorResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.RecvTensorResponse").msgclass
+  MarkRecvFinishedRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MarkRecvFinishedRequest").msgclass
+  MarkRecvFinishedResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MarkRecvFinishedResponse").msgclass
+  LoggingRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.LoggingRequest").msgclass
+  LabeledStepStats = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.LabeledStepStats").msgclass
+  LoggingResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.LoggingResponse").msgclass
+  TraceOpts = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.TraceOpts").msgclass
+  TracingRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.TracingRequest").msgclass
+  TracingResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.TracingResponse").msgclass
+  RecvBufRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.RecvBufRequest").msgclass
+  RecvBufResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.RecvBufResponse").msgclass
+  CompleteGroupRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CompleteGroupRequest").msgclass
+  CompleteGroupResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CompleteGroupResponse").msgclass
+  CompleteInstanceRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CompleteInstanceRequest").msgclass
+  CompleteInstanceResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.CompleteInstanceResponse").msgclass
+  GetStepSequenceRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GetStepSequenceRequest").msgclass
+  StepSequence = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.StepSequence").msgclass
+  GetStepSequenceResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.GetStepSequenceResponse").msgclass
+end

data/lib/tensorflow/core/protobuf/worker_service_pb.rb

@@ -0,0 +1,15 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/protobuf/worker_service.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/protobuf/worker_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/protobuf/worker_service.proto", :syntax => :proto3) do
+  end
+end
+
+module Tensorflow
+  module Grpc
+  end
+end

data/lib/tensorflow/core/util/event_pb.rb

@@ -0,0 +1,93 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/util/event.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/summary_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/util/event.proto", :syntax => :proto3) do
+    add_message "tensorflow.Event" do
+      optional :wall_time, :double, 1
+      optional :step, :int64, 2
+      oneof :what do
+        optional :file_version, :string, 3
+        optional :graph_def, :bytes, 4
+        optional :summary, :message, 5, "tensorflow.Summary"
+        optional :log_message, :message, 6, "tensorflow.LogMessage"
+        optional :session_log, :message, 7, "tensorflow.SessionLog"
+        optional :tagged_run_metadata, :message, 8, "tensorflow.TaggedRunMetadata"
+        optional :meta_graph_def, :bytes, 9
+      end
+    end
+    add_message "tensorflow.LogMessage" do
+      optional :level, :enum, 1, "tensorflow.LogMessage.Level"
+      optional :message, :string, 2
+    end
+    add_enum "tensorflow.LogMessage.Level" do
+      value :UNKNOWN, 0
+      value :DEBUGGING, 10
+      value :INFO, 20
+      value :WARN, 30
+      value :ERROR, 40
+      value :FATAL, 50
+    end
+    add_message "tensorflow.SessionLog" do
+      optional :status, :enum, 1, "tensorflow.SessionLog.SessionStatus"
+      optional :checkpoint_path, :string, 2
+      optional :msg, :string, 3
+    end
+    add_enum "tensorflow.SessionLog.SessionStatus" do
+      value :STATUS_UNSPECIFIED, 0
+      value :START, 1
+      value :STOP, 2
+      value :CHECKPOINT, 3
+    end
+    add_message "tensorflow.TaggedRunMetadata" do
+      optional :tag, :string, 1
+      optional :run_metadata, :bytes, 2
+    end
+    add_message "tensorflow.WatchdogConfig" do
+      optional :timeout_ms, :int64, 1
+    end
+    add_message "tensorflow.RequestedExitCode" do
+      optional :exit_code, :int32, 1
+    end
+    add_message "tensorflow.WorkerHeartbeatRequest" do
+      optional :shutdown_mode, :enum, 1, "tensorflow.WorkerShutdownMode"
+      optional :watchdog_config, :message, 2, "tensorflow.WatchdogConfig"
+      optional :exit_code, :message, 3, "tensorflow.RequestedExitCode"
+    end
+    add_message "tensorflow.WorkerHeartbeatResponse" do
+      optional :health_status, :enum, 1, "tensorflow.WorkerHealth"
+      repeated :worker_log, :message, 2, "tensorflow.Event"
+      optional :hostname, :string, 3
+    end
+    add_enum "tensorflow.WorkerHealth" do
+      value :OK, 0
+      value :RECEIVED_SHUTDOWN_SIGNAL, 1
+      value :INTERNAL_ERROR, 2
+      value :SHUTTING_DOWN, 3
+    end
+    add_enum "tensorflow.WorkerShutdownMode" do
+      value :DEFAULT, 0
+      value :NOT_CONFIGURED, 1
+      value :WAIT_FOR_COORDINATOR, 2
+      value :SHUTDOWN_AFTER_TIMEOUT, 3
+    end
+  end
+end
+
+module Tensorflow
+  Event = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.Event").msgclass
+  LogMessage = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.LogMessage").msgclass
+  LogMessage::Level = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.LogMessage.Level").enummodule
+  SessionLog = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.SessionLog").msgclass
+  SessionLog::SessionStatus = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.SessionLog.SessionStatus").enummodule
+  TaggedRunMetadata = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.TaggedRunMetadata").msgclass
+  WatchdogConfig = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.WatchdogConfig").msgclass
+  RequestedExitCode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.RequestedExitCode").msgclass
+  WorkerHeartbeatRequest = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.WorkerHeartbeatRequest").msgclass
+  WorkerHeartbeatResponse = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.WorkerHeartbeatResponse").msgclass
+  WorkerHealth = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.WorkerHealth").enummodule
+  WorkerShutdownMode = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.WorkerShutdownMode").enummodule
+end

data/lib/tensorflow/core/util/memmapped_file_system_pb.rb

@@ -0,0 +1,22 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/util/memmapped_file_system.proto
+
+require 'google/protobuf'
+
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/util/memmapped_file_system.proto", :syntax => :proto3) do
+    add_message "tensorflow.MemmappedFileSystemDirectoryElement" do
+      optional :offset, :uint64, 1
+      optional :name, :string, 2
+      optional :length, :uint64, 3
+    end
+    add_message "tensorflow.MemmappedFileSystemDirectory" do
+      repeated :element, :message, 1, "tensorflow.MemmappedFileSystemDirectoryElement"
+    end
+  end
+end
+
+module Tensorflow
+  MemmappedFileSystemDirectoryElement = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MemmappedFileSystemDirectoryElement").msgclass
+  MemmappedFileSystemDirectory = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.MemmappedFileSystemDirectory").msgclass
+end

data/lib/tensorflow/core/util/saved_tensor_slice_pb.rb

@@ -0,0 +1,40 @@
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: tensorflow/core/util/saved_tensor_slice.proto
+
+require 'google/protobuf'
+
+require 'tensorflow/core/framework/tensor_shape_pb'
+require 'tensorflow/core/framework/tensor_slice_pb'
+require 'tensorflow/core/framework/tensor_pb'
+require 'tensorflow/core/framework/types_pb'
+require 'tensorflow/core/framework/versions_pb'
+Google::Protobuf::DescriptorPool.generated_pool.build do
+  add_file("tensorflow/core/util/saved_tensor_slice.proto", :syntax => :proto3) do
+    add_message "tensorflow.SavedSliceMeta" do
+      optional :name, :string, 1
+      optional :shape, :message, 2, "tensorflow.TensorShapeProto"
+      optional :type, :enum, 3, "tensorflow.DataType"
+      repeated :slice, :message, 4, "tensorflow.TensorSliceProto"
+    end
+    add_message "tensorflow.SavedTensorSliceMeta" do
+      repeated :tensor, :message, 1, "tensorflow.SavedSliceMeta"
+      optional :versions, :message, 2, "tensorflow.VersionDef"
+    end
+    add_message "tensorflow.SavedSlice" do
+      optional :name, :string, 1
+      optional :slice, :message, 2, "tensorflow.TensorSliceProto"
+      optional :data, :message, 3, "tensorflow.TensorProto"
+    end
+    add_message "tensorflow.SavedTensorSlices" do
+      optional :meta, :message, 1, "tensorflow.SavedTensorSliceMeta"
+      optional :data, :message, 2, "tensorflow.SavedSlice"
+    end
+  end
+end
+
+module Tensorflow
+  SavedSliceMeta = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.SavedSliceMeta").msgclass
+  SavedTensorSliceMeta = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.SavedTensorSliceMeta").msgclass
+  SavedSlice = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.SavedSlice").msgclass
+  SavedTensorSlices = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("tensorflow.SavedTensorSlices").msgclass
+end

data/lib/tensorflow/data/batch_dataset.rb

@@ -0,0 +1,18 @@
+module Tensorflow
+  module Data
+    class BatchDataset < Dataset
+      def initialize(input_dataset, batch_size, drop_remainder)
+        @input_dataset = input_dataset
+        @output_types = input_dataset.output_types
+        @output_shapes = input_dataset.output_shapes.map do |shape|
+          shape.unshift(-1)
+        end
+
+        batch_size = Tensor.new(batch_size, dtype: :int64)
+        variant_tensor = RawOps.batch_dataset_v2(input_dataset.variant_tensor, batch_size, drop_remainder,
+                                                 output_types: @output_types, output_shapes: @output_shapes)
+        super(variant_tensor)
+      end
+    end
+  end
+end