Searched refs:kInputAttr (Results 1 – 2 of 2) sorted by relevance
| /external/libtextclassifier/native/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 | 57 constexpr const int kInputAttr = 3; variable
182 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Prepare()
186 TfLiteTensor& input = context->tensors[node->inputs->data[kInputAttr + i]]; in Prepare()
269 TF_LITE_ENSURE_EQ(context, node->inputs->size - kInputAttr, num_output_attrs); in Eval()
272 context->tensors[node->inputs->data[kInputAttr + i]], encoded_offsets, in Eval()
|