Searched refs:kEncoderInputRank (Results 1 – 4 of 4) sorted by relevance
33 constexpr const int kEncoderInputRank = 2; variable
36 TF_LITE_ENSURE_EQ(context, in.dims->size, kEncoderInputRank); in CopyValuesToTensorAndPadOrTruncate()
78 TF_LITE_ENSURE_EQ(context, num_tokens.dims->size, kEncoderInputRank); in Prepare()
166 TF_LITE_ENSURE_EQ(context, input_text.dims->size, kEncoderInputRank); in Prepare()