Home
last modified time | relevance | path

Searched refs:CreateIntArray (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/libtextclassifier/native/utils/tflite/
H A Dtext_encoder3s.cc68 CreateIntArray({kEncoderBatchSize, max_output_length}))); in ResizeOutputTensors()
75 CreateIntArray({kEncoderBatchSize, max_output_length}))); in ResizeOutputTensors()
84 CreateIntArray({kEncoderBatchSize, max_output_length}))); in ResizeOutputTensors()
110 CreateIntArray({kEncoderBatchSize}))); in Prepare()
H A Dencoder_common_test.cc25 TEST(EncoderUtilsTest, CreateIntArray) { in TEST() argument
26 TfLiteIntArray* a = CreateIntArray({1, 2, 3}); in TEST()
H A Dencoder_common.cc24 TfLiteIntArray* CreateIntArray(const std::initializer_list<int>& values) { in CreateIntArray() function
108 CreateIntArray({kEncoderBatchSize, max_output_length}))); in ResizeOutputTensor()
H A Dencoder_common.h37 TfLiteIntArray* CreateIntArray(const std::initializer_list<int>& values);
H A Dtoken_encoder.cc88 CreateIntArray({kEncoderBatchSize}))); in Prepare()
H A Dtext_encoder.cc178 CreateIntArray({kEncoderBatchSize}))); in Prepare()