Searched refs:segment_ids_tensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| H A D | bert_nl_classifier.cc | 75 auto* segment_ids_tensor = FindTensorByName( in Preprocess() local
123 PopulateTensor(std::vector<int>(input_tensor_length, 0), segment_ids_tensor); in Preprocess()
210 const auto& segment_ids_tensor = *FindTensorByName(input_tensors, input_tensor_metadatas, in InitializeFromMetadata() local
213 segment_ids_tensor.dims->size != 2) { in InitializeFromMetadata()
221 segment_ids_tensor.dims->size), in InitializeFromMetadata()
225 segment_ids_tensor.dims->data[0] != 1) { in InitializeFromMetadata()
233 segment_ids_tensor.dims->data[0]), in InitializeFromMetadata()
237 ids_tensor.dims->data[1] != segment_ids_tensor.dims->data[1]) { in InitializeFromMetadata()
244 segment_ids_tensor.dims->data[1]), in InitializeFromMetadata()
252 segment_ids_tensor.dims_signature->size != 2) { in InitializeFromMetadata()
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/testing/op_tests/ |
| H A D | unsorted_segment.py | 95 segment_ids_tensor = tf.constant(
102 output = unsorted_segment_op(data_tensor, segment_ids_tensor,
111 segment_ids_tensor = tf.compat.v1.placeholder(
120 intermediate_tensor = unsorted_segment_op(data_tensor, segment_ids_tensor,
131 return [data_tensor, segment_ids_tensor], [output]
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_dynamic_partition_op_test.py | 167 segment_ids_tensor = ragged_factory_ops.constant(
177 data_tensor, segment_ids_tensor, num_partitions)
185 data_flow_ops.dynamic_partition(data_tensor, segment_ids_tensor,
|
| /aosp_15_r20/external/tflite-support/tensorflow_lite_support/cc/task/text/qa/ |
| H A D | bert_question_answerer.cc | 176 TfLiteTensor* segment_ids_tensor = in Preprocess() local
281 PopulateTensor(segment_ids, segment_ids_tensor); in Preprocess()
|