int embedding size = 1;
 int lookup size = 1;
 for (int i = 0; i < lookup rank - 1; i++, k++) {
  const int dim = dense shape->data.i32[i];
  lookup size *= dim;
  output shape->data[k] = dim;
 }
 for (int i = 1; i < embedding rank; i++, k++) {
  const int dim = SizeOfDimension(value, i);
  embedding size *= dim;
  output shape->data[k] = dim;
 }