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

9 entries across 9 versions & 1 rubygems

Version	Path
tensor_stream-1.0.9	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-1.0.8	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-1.0.7	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-1.0.6	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-1.0.5	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-1.0.4	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-1.0.3	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-1.0.2	lib/tensor_stream/monkey_patches/float.rb
tensor_stream-1.0.1	lib/tensor_stream/monkey_patches/float.rb