Home
last modified time | relevance | path

Searched refs:current_total_weight (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dembedding_lookup_sparse.cc125 float current_total_weight, in FinalizeAggregation() argument
132 multiplier = current_total_weight; in FinalizeAggregation()
201 float current_total_weight = 0.0; in Eval() local
228 FinalizeAggregation(params->combiner, num_elements, current_total_weight, in Eval()
234 current_total_weight = 0.0; in Eval()
244 current_total_weight += w; in Eval()
252 FinalizeAggregation(params->combiner, num_elements, current_total_weight, in Eval()