Contents

class Float
  include TensorStream::MonkeyPatch

  def self.placeholder(name: nil, width: 32, shape: nil)
    raise "invalid width passed #{width}" unless [16, 32, 64].include?(width)

    data_type = :"float#{width}"
    TensorStream.placeholder(data_type, name: name, shape: shape)
  end
end

Version data entries

4 entries across 4 versions & 1 rubygems

Version	Path
tensor_stream-1.0.0	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-1.0.0.pre.rc1	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-0.9.10	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-0.9.9	lib/tensor_stream/monkey_patches/float.rb