deepfold.distributed.utils.GlobalMemoryBuffer

class deepfold.distributed.utils.GlobalMemoryBuffer[source]

Global buffer to avoid dynamic memory allocations.

Note

Caller should ensure that buffers of the same name are not used concurrently.

__init__()[source]

Methods

__init__()

get_tensor(tensor_shape, dtype, name)