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'