Lines Matching refs:tflite
30 class TokenEncoderOpModel : public tflite::SingleOpModel {
74 input_num_tokens_ = AddInput(tflite::TensorType_INT32); in TokenEncoderOpModel()
75 input_length_ = AddInput(tflite::TensorType_INT32); in TokenEncoderOpModel()
76 input_output_maxlength_ = AddInput(tflite::TensorType_INT32); in TokenEncoderOpModel()
77 input_attributes_int32_ = AddInput(tflite::TensorType_INT32); in TokenEncoderOpModel()
78 input_attributes_float_ = AddInput(tflite::TensorType_FLOAT32); in TokenEncoderOpModel()
80 output_positions_ = AddOutput(tflite::TensorType_INT32); in TokenEncoderOpModel()
81 output_length_ = AddOutput(tflite::TensorType_INT32); in TokenEncoderOpModel()
82 output_attributes_int32_ = AddOutput(tflite::TensorType_INT32); in TokenEncoderOpModel()
83 output_attributes_float_ = AddOutput(tflite::TensorType_FLOAT32); in TokenEncoderOpModel()
85 SetCustomOp("TokenEncoder", {}, tflite::ops::custom::Register_TOKEN_ENCODER); in TokenEncoderOpModel()