Searched defs:model_weights (Results 1 – 2 of 2) sorted by relevance
155 const int num_loss_partitions, const ModelWeights& model_weights, in ComputeWxAndWeightedExampleNorm()237 const ModelWeights& model_weights, in SampleAdaptiveProbabilities()
132 ModelWeights model_weights; in DoCompute() local