Searched defs:model_label (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/
H A Dpredict.cc81 TfLiteTensor* model_label = &context->tensors[node->inputs->data[2]]; local
85 TF_LITE_ENSURE_EQ(context, model_label->type, kTfLiteInt32);
89 TF_LITE_ENSURE_EQ(context, model_label->dims->size, 2);
92 model_label->dims->data[0]);
95 TF_LITE_ENSURE_EQ(context, model_label->dims->data[1],
119 TfLiteTensor* model_label = &context->tensors[node->inputs->data[2]]; local
126 const int items = model_label->dims->data[1];
135 aggregation[model_label->data.i32[idx * items + j]] +=

Completed in 79 milliseconds