Method: TensorStream::Train::SlotCreator#create_slot_with_initializer

Defined in:
lib/tensor_stream/train/slot_creator.rb

#create_slot_with_initializer(primary, initializer, shape, dtype, name, colocate_with_primary: true) ⇒ Object 



35
36
37
38
39
40
 
# File 'lib/tensor_stream/train/slot_creator.rb', line 35

def create_slot_with_initializer(primary, initializer, shape, dtype, name, colocate_with_primary: true)
  prefix = primary.op.name
  TensorStream.variable_scope(nil, prefix + "/" + name) do
    create_slot_var(primary, initializer, "", shape.shape)
  end
end