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

/external/tensorflow/tensorflow/contrib/lite/models/smartreply/ops/
H A Dpredict.cc106 TfLiteIntArray* weight_size = TfLiteIntArrayCreate(1); local
107 weight_size->data[0] = option->num_output;
113 return context->ResizeTensor(context, output_weight, weight_size);

Completed in 128 milliseconds