nvflare.app_opt.tensor_stream package

Submodules

• nvflare.app_opt.tensor_stream.client module
  • TensorClientStreamer
    • TensorClientStreamer.tensor_send_timeout
    • TensorClientStreamer.engine
    • TensorClientStreamer.sender
    • TensorClientStreamer.receiver
    • TensorClientStreamer.initialize()
    • TensorClientStreamer.handle_event()
    • TensorClientStreamer.send_tensors_to_server()
    • TensorClientStreamer.handle_event()
    • TensorClientStreamer.initialize()
    • TensorClientStreamer.send_tensors_to_server()
• nvflare.app_opt.tensor_stream.consumer module
  • TensorConsumer
    • TensorConsumer.logger
    • TensorConsumer.tensors_map
    • TensorConsumer.total_bytes
    • TensorConsumer.num_tensors
    • TensorConsumer.task_ids
    • TensorConsumer.consume()
    • TensorConsumer.finalize()
    • TensorConsumer.log_received()
    • TensorConsumer.process_shareable()
  • TensorConsumerFactory
    • TensorConsumerFactory.get_consumer()
    • TensorConsumerFactory.get_consumer()
• nvflare.app_opt.tensor_stream.producer module
  • TensorProducer
    • TensorProducer.logger
    • TensorProducer.tensor_send_timeout
    • TensorProducer.last
    • TensorProducer.tensors
    • TensorProducer.tensors_keys
    • TensorProducer.start
    • TensorProducer.current
    • TensorProducer.end
    • TensorProducer.produce()
    • TensorProducer.process_replies()
    • TensorProducer.log_completion()
    • TensorProducer.process_replies()
    • TensorProducer.produce()
• nvflare.app_opt.tensor_stream.receiver module
  • TensorReceiver
    • TensorReceiver.on_tensor_received()
    • TensorReceiver.set_ctx_with_tensors()
    • TensorReceiver.wait_for_tensors()
• nvflare.app_opt.tensor_stream.sender module
  • TensorSender
    • TensorSender.send()
    • TensorSender.store_tensors()
• nvflare.app_opt.tensor_stream.server module
  • TensorServerStreamer
    • TensorServerStreamer.tensor_send_timeout
    • TensorServerStreamer.engine
    • TensorServerStreamer.sender
    • TensorServerStreamer.receiver
    • TensorServerStreamer.start_sending_time
    • TensorServerStreamer.seen_tasks
    • TensorServerStreamer.num_task_data_sent
    • TensorServerStreamer.num_task_skipped
    • TensorServerStreamer.data_cleaned
    • TensorServerStreamer.lock
    • TensorServerStreamer.initialize()
    • TensorServerStreamer.handle_event()
    • TensorServerStreamer.send_tensors_to_client()
    • TensorServerStreamer.clean_counters()
    • TensorServerStreamer.wait_clients_to_complete()
    • TensorServerStreamer.try_to_clean_task_data()
    • TensorServerStreamer.clean_counters()
    • TensorServerStreamer.handle_event()
    • TensorServerStreamer.initialize()
    • TensorServerStreamer.send_tensors_to_client()
    • TensorServerStreamer.try_to_clean_task_data()
    • TensorServerStreamer.wait_sending_task_data_all_clients()
• nvflare.app_opt.tensor_stream.types module
  • TensorBlobKeys
    • TensorBlobKeys.PARENT_KEYS
    • TensorBlobKeys.SAFETENSORS_BLOB
    • TensorBlobKeys.TASK_ID
    • TensorBlobKeys.TENSOR_KEYS
  • TensorCustomKeys
    • TensorCustomKeys.SAFE_TENSORS_PROP_KEY
    • TensorCustomKeys.TASK_ID
  • TensorEventTypes
    • TensorEventTypes.SEND_TENSORS_FOR_TASK_DATA
  • TensorTopics
    • TensorTopics.TASK_DATA
    • TensorTopics.TASK_RESULT
• nvflare.app_opt.tensor_stream.utils module
  • chunk_tensors_from_params()
  • clean_task_data()
  • clean_task_result()
  • copy_non_tensor_params()
  • get_dxo_from_ctx()
  • get_targets_from_ctx_and_prop_key()
  • get_topic_for_ctx_prop_key()
  • merge_params_dicts()
  • to_numpy_recursive()
  • update_params_with_tensors()

Module contents