Searched refs:dims_signature (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.h | 186 const std::vector<int>* dims_signature = nullptr) {
189 if (dims_signature) {
190 rank_dims_signature = dims_signature->size();
191 dims_signature_pointer = dims_signature->data();
201 const int* dims_signature = nullptr);
|
| D | interpreter.cc | 338 const size_t rank_dims_signature, const int* dims_signature) { in SetTensorParametersReadWrite() argument
342 rank_dims_signature, dims_signature); in SetTensorParametersReadWrite()
|
| D | interpreter_builder.cc | 597 std::vector<int> dims_signature = {}; in ParseTensors() local
599 dims_signature = FlatBufferIntArrayToVector(tensor->shape_signature()); in ParseTensors()
631 dims_signature) != kTfLiteOk) { in ParseTensors()
|
| D | interpreter_test.cc | 583 std::vector<int> dims_signature = {-1, -1, 3}; in TEST() local
586 false, &dims_signature), in TEST()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.c | 147 if (t->dims_signature) { in TfLiteTensorFree()
148 TfLiteIntArrayFree((TfLiteIntArray *) t->dims_signature); in TfLiteTensorFree()
150 t->dims_signature = NULL; in TfLiteTensorFree()
|
| D | common_test.cc | 104 t.dims_signature = nullptr; in TEST()
120 t.dims_signature = nullptr; in TEST()
|
| D | common.h | 443 const TfLiteIntArray* dims_signature; member
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.h | 124 bool is_variable = false, const std::vector<int>& dims_signature = {}) {
125 if (dims_signature.empty()) {
132 is_variable, dims_signature.size(), dims_signature.data());
138 const int* dims_signature = nullptr);
|
| D | subgraph.cc | 853 if (tensor->dims_signature && tensor->dims_signature->size) { in ResizeInputTensorStrict()
854 dim_signature = tensor->dims_signature->data[idx]; in ResizeInputTensorStrict()
1255 const size_t rank_dims_signature, const int* dims_signature) { in SetTensorParametersReadWrite() argument
1297 tensor.dims_signature = in SetTensorParametersReadWrite()
1298 ConvertArrayToTfLiteIntArray(rank_dims_signature, dims_signature); in SetTensorParametersReadWrite()
|
| /external/tensorflow/tensorflow/lite/python/optimize/ |
| D | calibration_wrapper.cc | 273 if (tensor->dims_signature != nullptr && in SetTensor()
274 tensor->dims_signature->size == tensor->dims->size && in SetTensor()
275 tensor->dims_signature->data[j] == -1) { in SetTensor()
|
| /external/tensorflow/tensorflow/lite/java/src/main/native/ |
| D | tensor_jni.cc | 567 if (tensor->dims_signature != nullptr && tensor->dims_signature->size != 0) { in Java_org_tensorflow_lite_Tensor_shapeSignature()
568 num_dims = tensor->dims_signature->size; in Java_org_tensorflow_lite_Tensor_shapeSignature()
569 data = tensor->dims_signature->data; in Java_org_tensorflow_lite_Tensor_shapeSignature()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.cc | 352 if (tensor->dims_signature != nullptr && tensor->dims_signature->size != 0) { in TensorSizeSignature()
353 size_signature_data = tensor->dims_signature->data; in TensorSizeSignature()
354 size_signature_size = tensor->dims_signature->size; in TensorSizeSignature()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 385 if (tensor->dims_signature) { in HasUnspecifiedDimension()
386 for (int i : TfLiteIntArrayView(tensor->dims_signature)) { in HasUnspecifiedDimension()
1373 for (int i = 0; i < tensor->dims_signature->size; i++) { in AddTensor()
1374 dims_unspecified[i] = tensor->dims_signature->data[i] == -1 in AddTensor()
1376 : tensor->dims_signature->data[i]; in AddTensor()
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/configuration/ |
| D | configuration.proto | 128 // A tensor of with dynamic dimension must have a valid dims_signature
|