Contents

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

Version data entries

7 entries across 7 versions & 1 rubygems

Version	Path
tensor_stream-1.0.9	lib/tensor_stream/utils/data_type_utils.rb
tensor_stream-1.0.8	lib/tensor_stream/utils/data_type_utils.rb
tensor_stream-1.0.7	lib/tensor_stream/utils/data_type_utils.rb
tensor_stream-1.0.6	lib/tensor_stream/utils/data_type_utils.rb
tensor_stream-1.0.5	lib/tensor_stream/utils/data_type_utils.rb
tensor_stream-1.0.4	lib/tensor_stream/utils/data_type_utils.rb
tensor_stream-1.0.3	lib/tensor_stream/utils/data_type_utils.rb