Searched refs:weight_size (Results 1 – 2 of 2) sorted by relevance
107 TfLiteIntArray* weight_size = TfLiteIntArrayCreate(1); in Prepare() local108 weight_size->data[0] = option->num_output; in Prepare()114 return context->ResizeTensor(context, output_weight, weight_size); in Prepare()
157 const int64 weight_size = example_weights.size(); in Compute() local160 context, weight_size == 1 || weight_size == batch_size, in Compute()178 stream.PushEntry(feature_values(j), (weight_size > 1) in Compute()