Method: TensorStream::Graph#add_variable!

Defined in:
lib/tensor_stream/graph.rb

#add_variable!(node, options = {}) ⇒ Object 



181
182
183
184
185
186
 
# File 'lib/tensor_stream/graph.rb', line 181

def add_variable!(node, options = {})
  node = add_variable(node, options)
  op = Graph.get_default_graph.add_op!(:variable_v2, container: node, internal_name: node.name, shape: options[:shape], data_type: options[:data_type])
  node.name = op.name
  op
end