Searched refs:kEncoderBatchSize (Results 1 – 4 of 4) sorted by relevance
37 TF_LITE_ENSURE_EQ(context, in.dims->data[0], kEncoderBatchSize); in CopyValuesToTensorAndPadOrTruncate()97 CreateIntArray({kEncoderBatchSize, max_output_length}))); in ResizeOutputTensor()
34 constexpr const int kEncoderBatchSize = 1; variable
79 TF_LITE_ENSURE_EQ(context, num_tokens.dims->data[0], kEncoderBatchSize); in Prepare()88 CreateIntArray({kEncoderBatchSize}))); in Prepare()
167 TF_LITE_ENSURE_EQ(context, input_text.dims->data[0], kEncoderBatchSize); in Prepare()178 CreateIntArray({kEncoderBatchSize}))); in Prepare()