Class: TensorStream::DataTypeUtils

Inherits:
Object
  • Object
show all
Defined in:
lib/tensor_stream/utils/data_type_utils.rb

Class Method Summary collapse

Class Method Details

.norm_dtype(dtype) ⇒ Object 



3
4
5
6
7
8
9
10
11
12
13
 
# 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