Searched refs:output_bucket (Results 1 – 1 of 1) sorted by relevance
217 int output_bucket = 0; in Eval() local220 output_bucket += indices->data.i32[example_indices_offset + k] * stride; in Eval()223 const int output_offset = output_bucket * embedding_size; in Eval()