Module adseq.implementations.lossy_ring

Classes

class LossyRing (buffer: jax.Array)

Expand source code

class LossyRing(typing.NamedTuple):
    buffer: jax.Array
    @classmethod
    def init(cls, delay, capacity, grad=False):
        del delay
        return cls(jnp.full(capacity, INT_MAX,
            'float32' if grad else 'int32'))
    def enqueue(self, n):
        return _enqueue(self, n)
    def pop(self, n):
        return _pop(self, n)
    @classmethod
    def sized(cls, n):
        "wish I could use __class_getitem__"
        return type(f'{cls.__name__}[{n}]',
                    cls.__bases__,
                    {**cls.__dict__,
                     "init": functools.partial(cls.init, capacity=n)})

LossyRing(buffer,)

Ancestors

• builtins.tuple

Static methods

def init(delay, capacity, grad=False)

def sized(n)

wish I could use class_getitem

Instance variables

var buffer : jax.Array

Expand source code

class LossyRing(typing.NamedTuple):
    buffer: jax.Array
    @classmethod
    def init(cls, delay, capacity, grad=False):
        del delay
        return cls(jnp.full(capacity, INT_MAX,
            'float32' if grad else 'int32'))
    def enqueue(self, n):
        return _enqueue(self, n)
    def pop(self, n):
        return _pop(self, n)
    @classmethod
    def sized(cls, n):
        "wish I could use __class_getitem__"
        return type(f'{cls.__name__}[{n}]',
                    cls.__bases__,
                    {**cls.__dict__,
                     "init": functools.partial(cls.init, capacity=n)})

Alias for field number 0

Methods

def enqueue(self, n)

Expand source code

def enqueue(self, n):
    return _enqueue(self, n)

def pop(self, n)

Expand source code

def pop(self, n):
    return _pop(self, n)