Searched refs:kInputAttr (Results 1 – 2 of 2) sorted by relevance
| /external/libtextclassifier/utils/tflite/ |
| D | token_encoder.cc | 40 constexpr const int kInputAttr = 3; variable
92 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Prepare()
96 TfLiteTensor& input = context->tensors[node->inputs->data[kInputAttr + i]]; in Prepare()
160 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Eval()
163 context->tensors[node->inputs->data[kInputAttr + i]], encoded_offsets, in Eval()
|
| D | text_encoder.cc | 56 constexpr const int kInputAttr = 3; variable
181 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Prepare()
185 TfLiteTensor& input = context->tensors[node->inputs->data[kInputAttr + i]]; in Prepare()
268 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Eval()
271 context->tensors[node->inputs->data[kInputAttr + i]], encoded_offsets, in Eval()
|