tensor_stream 1.0.2 → 1.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c66c4a398833de1008fefcd0f9392b194a930f884ed494769a27e524450ab290
-  data.tar.gz: 334ec92a8608270d9ffc1f34cc65591753a68fc8b7ddb4ea0ded8ad923264e6c
+  metadata.gz: 408dd69cf3211817d31734564f9a51c9ff9fdfca9ab7f207994f67b2b6d82303
+  data.tar.gz: f07f0b3ebc776dbd1e836968d5e8c97b9730519c6d57251f14cf7a08581b77a3
 SHA512:
-  metadata.gz: 92de43776651b67337a701aa89f25738c476c20fc389b51599d1a6038345cb0f08a7b0a7bd122d57f64be368292d41debeb278d784add5bfff5f7a1261cd8c62
-  data.tar.gz: 393e9e2ff6467e88174c0b199b025beddec8cb0738da42711686818684df7628cdaff402b9227d934193301f52e9da314aa2e4f421b052750785e3df784d158e
+  metadata.gz: '079a28b96e99180438059036db170dccbb8437188893031a607fcdebc8e66d6ea6678f1da83bd2bf2a8515434e061867e02f9342d00f5a8aadbceb141465ed09'
+  data.tar.gz: 1f8556098e4fa005a36e2884a079b9ea862aeef0c077623215430b69201a6e7e56b3fd4e33036f54a5c554cbf17fffe71a792f1984f4ac6fbcb116a14104453e

data/lib/tensor_stream.rb

@@ -2,6 +2,7 @@ require "tensor_stream/version"
 require "deep_merge"
 require "matrix"
 require "concurrent"
+require "tensor_stream/utils/data_type_utils"
 require "tensor_stream/exceptions"
 require "tensor_stream/helpers/op_helper"
 require "tensor_stream/helpers/string_helper"

data/lib/tensor_stream/placeholder.rb

@@ -4,7 +4,7 @@ module TensorStream
     def initialize(data_type, rank, shape, options = {})
       setup_initial_state(options)
 
-      @data_type = data_type.to_sym
+      @data_type = DataTypeUtils.norm_dtype(data_type.to_sym)
       @rank = rank
       @shape = TensorShape.new(shape, rank)
       @value = nil

data/lib/tensor_stream/utils.rb

@@ -279,7 +279,7 @@ module TensorStream
     end
 
     def check_data_types(*args)
-      unique_types = args.select { |a| a.is_a?(Tensor) }. map { |a| norm_dtype(a.data_type) }.uniq
+      unique_types = args.select { |a| a.is_a?(Tensor) }. map { |a| DataTypeUtils.norm_dtype(a.data_type) }.uniq
 
       if unique_types.size > 1
         raise TensorStream::ValueError, "Value Error: Tensor conversion requested dtypes are different -> #{unique_types}"
@@ -295,17 +295,5 @@ module TensorStream
       coerced_type = check_data_types(*args)
       args.map { |a| a.is_a?(Tensor) ? a : convert_to_tensor(a, dtype: coerced_type) }
     end
-
-    def norm_dtype(dtype)
-      dtype = dtype.to_sym
-      case dtype
-      when :int
-        :int32
-      when :float
-        :float32
-      else
-        dtype
-      end
-    end
   end
 end

data/lib/tensor_stream/utils/data_type_utils.rb

@@ -0,0 +1,15 @@
+module TensorStream
+  class DataTypeUtils
+    def self.norm_dtype(dtype)
+      dtype = dtype.to_sym
+      case dtype
+      when :int
+        :int32
+      when :float
+        :float32
+      else
+        dtype
+      end
+    end
+  end
+end

data/lib/tensor_stream/version.rb

@@ -1,5 +1,5 @@
 module TensorStream
-  VERSION = "1.0.2".freeze
+  VERSION = "1.0.3".freeze
 
   def self.version
     VERSION

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tensor_stream
 version: !ruby/object:Gem::Version
-  version: 1.0.2
+  version: 1.0.3
 platform: ruby
 authors:
 - Joseph Emmanuel Dayo
@@ -376,6 +376,7 @@ files:
 - lib/tensor_stream/trainer.rb
 - lib/tensor_stream/types.rb
 - lib/tensor_stream/utils.rb
+- lib/tensor_stream/utils/data_type_utils.rb
 - lib/tensor_stream/utils/freezer.rb
 - lib/tensor_stream/variable.rb
 - lib/tensor_stream/variable_scope.rb