IMLCV.implementations.tensorflow.CV

IMLCV.implementations.tensorflow.CV#

Classes#

tfl_module

Base class for dataclasses that should act like a JAX pytree node.

KerasFunBase

Helper class that provides a standard way to create an ABC using

Module Contents#

class IMLCV.implementations.tensorflow.CV.tfl_module(*args, **kwargs)#

Bases: IMLCV.base.datastructures.MyPyTreeNode

Base class for dataclasses that should act like a JAX pytree node.

mod: tensorflow.Module#
get_config()#
__getstate__()#
__setstate__(state)#
class IMLCV.implementations.tensorflow.CV.KerasFunBase(*args, **kwargs)#

Bases: IMLCV.base.CV.CvFunBase

Helper class that provides a standard way to create an ABC using inheritance.

fwd: tfl_module#
bwd: tfl_module | None#
static create(fwd: tensorflow.Module, bwd: tensorflow.Module = None)#
_calc(x: IMLCV.base.CV.CV, nl: IMLCV.base.CV.NeighbourList, reverse=False, conditioners: list[IMLCV.base.CV.CV] | None = None, shmap=False) → IMLCV.base.CV.CV#
__getstate__()#
__setstate__(state)#
Contents