Home
last modified time | relevance | path

Searched refs:model_tensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/sentencepiece/
H A Dsentencepiece_detokenizer_tflite.cc55 const TfLiteTensor& model_tensor = in Eval() local
57 const auto model_buffer_data = model_tensor.data.data; in Eval()
H A Dsentencepiece_detokenizer_op.cc49 const auto& model_tensor = ctx->input(kSPModelIndex); in Compute() local
71 codes_for_split, model_tensor.data()); in Compute()
H A Dsentencepiece_tokenizer_op.cc67 const auto& model_tensor = ctx->input(kSPModelIndex); in Compute() local
84 input_values_flat(i), model_tensor.data(), add_bos, add_eos, reverse); in Compute()
H A Dsentencepiece_tokenizer_tflite.cc69 const TfLiteTensor& model_tensor = in Eval() local
71 const auto model_buffer_data = model_tensor.data.data; in Eval()