Searched refs:lookup_rank (Results 1 – 1 of 1) sorted by relevance
162 const int lookup_rank = SizeOfDimension(indices, 1); in Eval() local168 const int output_rank = (lookup_rank - 1) + (embedding_rank - 1); in Eval()172 TF_LITE_ENSURE_EQ(context, SizeOfDimension(dense_shape, 0), lookup_rank); in Eval()179 for (int i = 0; i < lookup_rank - 1; i++, k++) { in Eval()216 const int example_indices_offset = i * lookup_rank; in Eval()219 for (int k = (lookup_rank - 1) - 1; k >= 0; k--) { in Eval()