Searched refs:weight_index (Results 1 – 7 of 7) sorted by relevance
109 int WeightedPicker::PickAt(int32_t weight_index) const { in PickAt()110 if (weight_index < 0 || weight_index >= total_weight()) return -1; in PickAt()112 int32_t position = weight_index; in PickAt()
226 int weight_index = 0; in TestPickAt() local229 int pick = picker.PickAt(weight_index); in TestPickAt()231 ++weight_index; in TestPickAt()234 EXPECT_EQ(weight_index, picker.total_weight()); in TestPickAt()
57 int PickAt(int32_t weight_index) const;
194 const int weight_index = weight_start + frequency_offset + j; in FilterbankPopulateState() local195 QuantizeFilterbankWeights(weight, state->weights + weight_index, in FilterbankPopulateState()196 state->unweights + weight_index); in FilterbankPopulateState()
327 std::string weight_index = "ky"; in GenerateCode() local329 weight_index = in GenerateCode()332 weight_value = "weights_cache[" + weight_index + " * " + in GenerateCode()
1285 weight_index = 01290 tensors = weights[weight_index:weight_index + num_tensors]1292 weight_index += num_tensors1294 weight = weights[weight_index]1302 weight_index += 1
1783 weight_index = 01788 tensors = weights[weight_index:weight_index + num_tensors]1790 weight_index += num_tensors1792 weight = weights[weight_index]1800 weight_index += 1