Searched refs:SetWeightsFromArray (Results 1 – 3 of 3) sorted by relevance
144 picker.SetWeightsFromArray(size, &weights[0]); in TestPicker()178 array_picker.SetWeightsFromArray(size, &weights[0]); in TestPicker()225 picker.SetWeightsFromArray(items, weights); in TestPickAt()251 p.SetWeightsFromArray(arg, &weights[0]); in BM_CreateAndSetWeights()
81 void SetWeightsFromArray(int N, const int32* weights);
155 void WeightedPicker::SetWeightsFromArray(int N, const int32* weights) { in SetWeightsFromArray() function in tensorflow::random::WeightedPicker