Method: Tensorflow::Graph::FunctionDef#process_signatures

Defined in:
lib/tensorflow/graph/function_def.rb

#process_signatures(ruby_method, input_signatures) ⇒ Object 



14
15
16
17
18
19
20
21
22
 
# File 'lib/tensorflow/graph/function_def.rb', line 14

def process_signatures(ruby_method, input_signatures)
  if input_signatures.length != ruby_method.parameters.length
    raise(Error::InvalidArgumentError, "Must specify input signature for each method parameter")
  end

  @signatures = input_signatures.map do |dtype, shape|
    Signature.new(dtype, shape)
  end
end