Home
last modified time | relevance | path

Searched refs:num_output_attrs (Results 1 – 2 of 2) sorted by relevance

/external/libtextclassifier/utils/tflite/
Dtoken_encoder.cc63 const int num_output_attrs = node->outputs->size - kOutputAttr; in ResizeOutputTensors() local
64 for (int i = 0; i < num_output_attrs; ++i) { in ResizeOutputTensors()
91 const int num_output_attrs = node->outputs->size - kOutputAttr; in Prepare() local
92 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Prepare()
95 for (int i = 0; i < num_output_attrs; ++i) { in Prepare()
109 for (int i = 0; i < num_output_attrs; ++i) { in Prepare()
159 const int num_output_attrs = node->outputs->size - kOutputAttr; in Eval() local
160 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Eval()
161 for (int i = 0; i < num_output_attrs; ++i) { in Eval()
Dtext_encoder.cc150 const int num_output_attrs = node->outputs->size - kOutputAttr; in ResizeOutputTensors() local
151 for (int i = 0; i < num_output_attrs; ++i) { in ResizeOutputTensors()
180 const int num_output_attrs = node->outputs->size - kOutputAttr; in Prepare() local
181 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Prepare()
184 for (int i = 0; i < num_output_attrs; ++i) { in Prepare()
199 for (int i = 0; i < num_output_attrs; ++i) { in Prepare()
267 const int num_output_attrs = node->outputs->size - kOutputAttr; in Eval() local
268 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Eval()
269 for (int i = 0; i < num_output_attrs; ++i) { in Eval()