spark.core.cache

Contents

spark.core.cache#

Classes#

Cache dataclass.

Module Contents#

class spark.core.cache.Cache(variable, payload_type)[source]#

Cache dataclass.

Parameters:

variable (spark.core.variables.Variable)
payload_type (type[spark.core.payloads.ValueSparkPayload])

variable: spark.core.variables.Variable[source]#

payload_type: type[spark.core.payloads.ValueSparkPayload][source]#

property value: spark.core.payloads.ValueSparkPayload[source]#

Current value store in the cache object.

Return type:: spark.core.payloads.ValueSparkPayload

property shape: tuple[int, Ellipsis][source]#

Shape of the value store in the cache object.

Return type:: tuple[int, Ellipsis]

property dtype: jax.numpy.dtype[source]#

Dtype of the value store in the cache object.

Return type:: jax.numpy.dtype