Sha256: ffc2559ce539bec1c3cde4c832e175109e921ee120a545d92cfca3e21436acae
Contents?: true
Size: 586 Bytes
Versions: 19
Compression:
Stored size: 586 Bytes
Contents
module TensorStream
# Class that defines a TensorStream placeholder
class Placeholder < Tensor
def initialize(data_type, rank, shape, options = {})
setup_initial_state(options)
@data_type = data_type.to_sym
@rank = rank
@shape = TensorShape.new(shape, rank)
@value = nil
@is_const = false
@name = [@graph.get_name_scope, options[:name] || build_name].compact.reject(&:empty?).join('/')
@graph.add_node(self)
end
private
def build_name
"Placeholder#{graph.get_placeholder_counter}:#{@rank}"
end
end
end
Version data entries
19 entries across 19 versions & 1 rubygems