Module: TensorFlow::Utils

Included in:
TensorFlow
Defined in:
lib/tensorflow/utils.rb

Class Method Summary collapse

Class Method Details

.check_status(status) ⇒ Object 



3
4
5
6
7
 
# File 'lib/tensorflow/utils.rb', line 3

def self.check_status(status)
  if FFI.TF_GetCode(status) != 0
    raise Error, FFI.TF_Message(status)
  end
end

.infer_type(value) ⇒ Object 



9
10
11
12
13
14
15
16
17
18
19
20
 
# File 'lib/tensorflow/utils.rb', line 9

def self.infer_type(value)
  case value
  when String
    :string
  when Float
    :float
  when true, false
    :bool
  else
    :int32
  end
end