class TensorStream::DataTypeUtils

Public Class Methods

norm_dtype(dtype) click to toggle source
# File lib/tensor_stream/utils/data_type_utils.rb, line 3
def self.norm_dtype(dtype)
  dtype = dtype.to_sym
  case dtype
  when :int
    :int32
  when :float
    :float32
  else
    dtype
  end
end