openfl.protocols.utils.construct_model_proto

openfl.protocols.utils.construct_model_proto(tensor_dict, round_number, tensor_pipe)

Construct model proto from tensor dict.

Parameters:

• tensor_dict – A dictionary where the keys are tensor names and the values are the corresponding tensors.

• round_number – The round number for the model.

• tensor_pipe – The tensor pipe for the model.

Returns:

model_proto – The protobuf of the model.