Home
last modified time | relevance | path

Searched defs:ids_tensor (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Ddictionary_ops.cc148 tf::Tensor* ids_tensor; in DoCompute() local
203 const tf::Tensor& ids_tensor = context->input(0); in DoCompute() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/
H A Dbert_nl_classifier.cc71 auto* ids_tensor = in Preprocess() local
206 const auto& ids_tensor = *FindTensorByName(input_tensors, input_tensor_metadatas, in InitializeFromMetadata() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/
H A Dbert_question_answerer.cc166 TfLiteTensor* ids_tensor = in Preprocess() local