| /external/tflite-support/tensorflow_lite_support/cc/task/text/nlclassifier/ |
| D | bert_nl_classifier.cc | 250 if (ids_tensor.dims_signature->size != 2 || in InitializeFromMetadata()
251 mask_tensor.dims_signature->size != 2 || in InitializeFromMetadata()
252 segment_ids_tensor.dims_signature->size != 2) { in InitializeFromMetadata()
256 if (ids_tensor.dims_signature->data[1] == -1 && in InitializeFromMetadata()
257 mask_tensor.dims_signature->data[1] == -1 && in InitializeFromMetadata()
258 segment_ids_tensor.dims_signature->data[1] == -1) { in InitializeFromMetadata()
260 } else if (ids_tensor.dims_signature->data[1] == -1 || in InitializeFromMetadata()
261 mask_tensor.dims_signature->data[1] == -1 || in InitializeFromMetadata()
262 segment_ids_tensor.dims_signature->data[1] == -1) { in InitializeFromMetadata()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common_test.cc | 108 t.dims_signature = nullptr; in TEST()
124 t.dims_signature = nullptr; in TEST()
167 src.dims_signature = TfLiteIntArrayCopy(src.dims); in TEST()
|
| D | common.cc | 172 if (t->dims_signature) { in TfLiteTensorFree()
173 TfLiteIntArrayFree((TfLiteIntArray*)t->dims_signature); in TfLiteTensorFree()
175 t->dims_signature = nullptr; in TfLiteTensorFree()
|
| D | common.h | 466 const TfLiteIntArray* dims_signature; member
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 131 bool is_variable = false, const std::vector<int>& dims_signature = {}) {
132 if (dims_signature.empty()) {
139 is_variable, dims_signature.size(), dims_signature.data());
145 const int* dims_signature = nullptr);
|
| D | subgraph.cc | 935 if (tensor->dims_signature && tensor->dims_signature->size) { in ResizeInputTensorStrict()
936 dim_signature = tensor->dims_signature->data[idx]; in ResizeInputTensorStrict()
1510 const size_t ndims_signature, const int* dims_signature) { in SetTensorParametersReadWrite() argument
1551 tensor.dims_signature = in SetTensorParametersReadWrite()
1552 ConvertArrayToTfLiteIntArray(ndims_signature, dims_signature); in SetTensorParametersReadWrite()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.h | 193 const std::vector<int>* dims_signature = nullptr) {
196 if (dims_signature) {
197 rank_dims_signature = dims_signature->size();
198 dims_signature_pointer = dims_signature->data();
208 const int* dims_signature = nullptr);
|
| D | interpreter.cc | 280 const size_t rank_dims_signature, const int* dims_signature) { in SetTensorParametersReadWrite() argument
284 rank_dims_signature, dims_signature); in SetTensorParametersReadWrite()
|
| D | interpreter_builder.cc | 625 std::vector<int> dims_signature = {}; in ParseTensors() local
627 dims_signature = FlatBufferIntArrayToVector(tensor->shape_signature()); in ParseTensors()
659 dims_signature) != kTfLiteOk) { in ParseTensors()
|
| D | interpreter_test.cc | 589 std::vector<int> dims_signature = {-1, -1, 3}; in TEST() local
592 false, &dims_signature), in TEST()
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper.cc | 424 if (tensor->dims_signature != nullptr && in SetTensor()
425 tensor->dims_signature->size == tensor->dims->size && in SetTensor()
426 tensor->dims_signature->data[j] == -1) { in SetTensor()
504 if (tensor->dims_signature != nullptr && in SetTensor()
505 tensor->dims_signature->size == tensor->dims->size && in SetTensor()
506 tensor->dims_signature->data[j] == -1) { in SetTensor()
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib_test.cc | 356 std::vector<int> dims_signature = {-1, 3}; in TEST_P() local
359 /*is_variable=*/false, &dims_signature); in TEST_P()
362 /*is_variable=*/false, &dims_signature); in TEST_P()
365 /*is_variable=*/false, &dims_signature); in TEST_P()
391 CHECK_NOTNULL(tensor0->dims_signature); in TEST_P()
392 TfLiteIntArrayView shape_view(tensor0->dims_signature); in TEST_P()
|
| D | writer_lib.cc | 255 if (tensor->dims_signature != nullptr) { in ExportTensors()
256 TfLiteIntArrayView shape_signature_view(tensor->dims_signature); in ExportTensors()
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | tensor_jni.cc | 680 if (tensor->dims_signature != nullptr && tensor->dims_signature->size != 0) { in Java_org_tensorflow_lite_TensorImpl_shapeSignature()
681 num_dims = tensor->dims_signature->size; in Java_org_tensorflow_lite_TensorImpl_shapeSignature()
682 data = tensor->dims_signature->data; in Java_org_tensorflow_lite_TensorImpl_shapeSignature()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util_test.cc | 200 tflite_tensor.dims_signature = nullptr; in TEST()
237 tflite_tensor.dims_signature = nullptr; in TEST()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.cc | 423 if (tensor->dims_signature != nullptr && tensor->dims_signature->size != 0) { in TensorSizeSignature()
424 size_signature_data = tensor->dims_signature->data; in TensorSizeSignature()
425 size_signature_size = tensor->dims_signature->size; in TensorSizeSignature()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 584 if (tensor->dims_signature) { in HasUnspecifiedDimension()
585 for (int i : TfLiteIntArrayView(tensor->dims_signature)) { in HasUnspecifiedDimension()
|
| D | kernel_util_test.cc | 910 tensor.dims_signature = shape_sig; in TEST_F()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 1843 for (int i = 0; i < tensor->dims_signature->size; i++) { in AddTensor()
1844 dims_unspecified[i] = tensor->dims_signature->data[i] == -1 in AddTensor()
1846 : tensor->dims_signature->data[i]; in AddTensor()
2173 if (tensor.dims_signature) { in AppendDynamicDimensions()
2174 for (int i = 0; i < tensor.dims_signature->size; i++) { in AppendDynamicDimensions()
2175 if (tensor.dims_signature->data[i] == -1) { in AppendDynamicDimensions()
|
| D | nnapi_delegate_test.cc | 5504 if (tensor->dims_signature != nullptr && in GetNNAPIDimensions()
5505 tensor->dims_signature->size == tensor->dims->size) { in GetNNAPIDimensions()
5506 for (auto d : TfLiteIntArrayView(tensor->dims_signature)) { in GetNNAPIDimensions()
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/ |
| D | configuration.proto | 137 // A tensor of with dynamic dimension must have a valid dims_signature
|