Searched refs:ids_tensor (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| H A D | bert_nl_classifier.cc | 71 auto* ids_tensor = in Preprocess() local
121 PopulateTensor(input_ids, ids_tensor); in Preprocess()
206 const auto& ids_tensor = *FindTensorByName(input_tensors, input_tensor_metadatas, in InitializeFromMetadata() local
212 if (ids_tensor.dims->size != 2 || mask_tensor.dims->size != 2 || in InitializeFromMetadata()
220 ids_tensor.dims->size, mask_tensor.dims->size, in InitializeFromMetadata()
224 if (ids_tensor.dims->data[0] != 1 || mask_tensor.dims->data[0] != 1 || in InitializeFromMetadata()
232 ids_tensor.dims->data[0], mask_tensor.dims->data[0], in InitializeFromMetadata()
236 if (ids_tensor.dims->data[1] != mask_tensor.dims->data[1] || in InitializeFromMetadata()
237 ids_tensor.dims->data[1] != segment_ids_tensor.dims->data[1]) { in InitializeFromMetadata()
243 ids_tensor.dims->data[1], mask_tensor.dims->data[1], in InitializeFromMetadata()
[all …]
|
| /aosp_15_r20/external/federated-compute/fcp/tensorflow/ |
| H A D | dictionary_ops.cc | 148 tf::Tensor* ids_tensor; in DoCompute() local
150 context->allocate_output(0, token_tensor.shape(), &ids_tensor); in DoCompute()
162 if (ids_tensor->dtype() != tf::DataType::DT_INT64) { in DoCompute()
165 if (token_tensor.shape() != ids_tensor->shape()) { in DoCompute()
169 auto ids_flat = ids_tensor->flat<int64_t>(); in DoCompute()
203 const tf::Tensor& ids_tensor = context->input(0); in DoCompute() local
206 context->allocate_output(0, ids_tensor.shape(), &token_tensor); in DoCompute()
218 if (ids_tensor.dtype() != tf::DataType::DT_INT64) { in DoCompute()
221 if (ids_tensor.shape() != token_tensor->shape()) { in DoCompute()
225 const auto ids_flat = ids_tensor.flat<int64_t>(); in DoCompute()
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/ |
| H A D | bert_question_answerer.cc | 166 TfLiteTensor* ids_tensor = in Preprocess() local
277 PopulateTensor(input_ids, ids_tensor); in Preprocess()
|