Home
last modified time | relevance | path

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

/external/libtextclassifier/native/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.cc151 const int num_output_attrs = node->outputs->size - kOutputAttr; in ResizeOutputTensors() local
152 for (int i = 0; i < num_output_attrs; ++i) { in ResizeOutputTensors()
181 const int num_output_attrs = node->outputs->size - kOutputAttr; in Prepare() local
182 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Prepare()
185 for (int i = 0; i < num_output_attrs; ++i) { in Prepare()
200 for (int i = 0; i < num_output_attrs; ++i) { in Prepare()
268 const int num_output_attrs = node->outputs->size - kOutputAttr; in Eval() local
269 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Eval()
270 for (int i = 0; i < num_output_attrs; ++i) { in Eval()