Searched refs:output_weight (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/
H A Dpredict.cc100 TfLiteTensor* output_weight = &context->tensors[node->outputs->data[1]]; local
102 TF_LITE_ENSURE_EQ(context, output_weight->type, kTfLiteFloat32);
113 return context->ResizeTensor(context, output_weight, weight_size);
148 TfLiteTensor* output_weight = &context->tensors[node->outputs->data[1]]; local
154 output_weight->data.f[i] = 0.0f;
157 output_weight->data.f[i] = sorted_labels[i].second;

Completed in 142 milliseconds