spark.core.cache

Contents

spark.core.cache#

Classes#

Cache dataclass.

Module Contents#

class spark.core.cache.Cache(_variable)[source]#

Cache dataclass.

Parameters:: _variable (spark.core.payloads.SparkPayload)

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

Current value store in the cache object.

Return type:: spark.core.payloads.SparkPayload

tree_flatten()[source]#

Return type:: tuple[tuple, tuple]

classmethod tree_unflatten(aux_data, children)[source]#

Return type:: Self

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

get()[source]#

Return type:: spark.core.payloads.SparkPayload

set(payload)[source]#

Parameters:: payload (spark.core.payloads.SparkPayload)
Return type:: None