nvflare.app_opt.tensor_stream.types module

class TensorBlobKeys[source]

Bases: object

PARENT_KEYS = 'parent_keys'
SAFETENSORS_BLOB = 'safetensors_blob'
TASK_ID = 'task_id'
TENSOR_KEYS = 'tensor_keys'
class TensorCustomKeys[source]

Bases: object

SAFE_TENSORS_PROP_KEY = '_safe_tensors_blob_'
TASK_ID = 'task_id'
class TensorEventTypes[source]

Bases: object

SEND_TENSORS_FOR_TASK_DATA = 'SEND_TENSORS_FOR_TASK_DATA'
class TensorTopics[source]

Bases: object

TASK_DATA = 'task_data'
TASK_RESULT = 'task_result'