Contents

module TensorStream
  module Images
    extend OpHelper
    extend TensorStream::Utils

    def self.decode_png(contents, channels: 0, dtype: :uint8, name: nil)
      _op(:decode_png, contents, channels: channels, data_type: dtype, name: name)
    end

    def self.encode_png(contents, compression: -1, name: nil)
      check_allowed_types(contents, [:uint8, :uint16])
      contents = convert_to_tensor(contents, dtype: :uint16)
      _op(:encode_png, contents, compression: compression, name: name)
    end
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
tensor_stream-0.9.2	lib/tensor_stream/images.rb
tensor_stream-0.9.1	lib/tensor_stream/images.rb
tensor_stream-0.9.0	lib/tensor_stream/images.rb
tensor_stream-0.8.6	lib/tensor_stream/images.rb
tensor_stream-0.8.5	lib/tensor_stream/images.rb