Home
last modified time | relevance | path

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

/external/libtextclassifier/utils/tflite/
Dtoken_encoder.cc53 constexpr const int kOutputAttr = 2; variable
63 const int num_output_attrs = node->outputs->size - kOutputAttr; in ResizeOutputTensors()
69 &context->tensors[node->outputs->data[kOutputAttr + i]], context)); in ResizeOutputTensors()
91 const int num_output_attrs = node->outputs->size - kOutputAttr; in Prepare()
98 context->tensors[node->outputs->data[kOutputAttr + i]]; in Prepare()
111 context->tensors[node->outputs->data[kOutputAttr + i]]; in Prepare()
159 const int num_output_attrs = node->outputs->size - kOutputAttr; in Eval()
165 &context->tensors[node->outputs->data[kOutputAttr + i]]); in Eval()
Dtext_encoder.cc72 constexpr const int kOutputAttr = 3; variable
150 const int num_output_attrs = node->outputs->size - kOutputAttr; in ResizeOutputTensors()
156 &context->tensors[node->outputs->data[kOutputAttr + i]], context)); in ResizeOutputTensors()
180 const int num_output_attrs = node->outputs->size - kOutputAttr; in Prepare()
187 context->tensors[node->outputs->data[kOutputAttr + i]]; in Prepare()
201 context->tensors[node->outputs->data[kOutputAttr + i]]; in Prepare()
267 const int num_output_attrs = node->outputs->size - kOutputAttr; in Eval()
273 &context->tensors[node->outputs->data[kOutputAttr + i]]); in Eval()