Searched refs:broadcast_reshaped_weights (Results 1 – 1 of 1) sorted by relevance
289 auto broadcast_reshaped_weights = xla::BroadcastInDim( in CalculateGradData() local299 broadcast_grad_output * broadcast_reshaped_weights; in CalculateGradData()