[go: nahoru, domu]

Skip to content

Commit

Permalink
Fix variable name.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 644525699
  • Loading branch information
impjdi authored and tensorflower-gardener committed Jun 18, 2024
1 parent 1722894 commit df114d8
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions tensorflow/lite/kernels/embedding_lookup.cc
Original file line number Diff line number Diff line change
Expand Up @@ -83,14 +83,14 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) {

TfLiteTensor* output;
TF_LITE_ENSURE_OK(context, GetOutputSafe(context, node, 0, &output));
TfLiteIntArray* outputSize = TfLiteIntArrayCreate(NumDimensions(value));
TfLiteIntArray* output_size = TfLiteIntArrayCreate(NumDimensions(value));

outputSize->data[0] = SizeOfDimension(lookup, 0);
outputSize->data[1] = SizeOfDimension(value, 1);
output_size->data[0] = SizeOfDimension(lookup, 0);
output_size->data[1] = SizeOfDimension(value, 1);
for (int i = 2; i < NumDimensions(value); i++) {
outputSize->data[i] = SizeOfDimension(value, i);
output_size->data[i] = SizeOfDimension(value, i);
}
return context->ResizeTensor(context, output, outputSize);
return context->ResizeTensor(context, output, output_size);
}

TfLiteStatus EvalSimple(TfLiteContext* context, TfLiteNode* node,
Expand Down

0 comments on commit df114d8

Please sign in to comment.