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