Broadcast the model
Broadcast the model
SparkContext
model to broadcast
this
Get the broadcast model on worker
Get the broadcast model on worker
If create a tensor for gradient when fetch the model. Please note that the gradient is not needed in model inference
model
ModelBroadcast is used to broadcast model