class Tensorflow::Decorator::Function

Attributes

input_signatures[R]

Public Class Methods

new(input_signatures = []) click to toggle source
# File lib/tensorflow/decorators.rb, line 6
def initialize(input_signatures = [])
  @input_signatures = input_signatures
end

Public Instance Methods

wrap(method) click to toggle source
# File lib/tensorflow/decorators.rb, line 10
def wrap(method)
  Graph::FunctionDef.new(method, self.input_signatures)
end