| /external/armnn/src/armnnTfLiteParser/test/ |
| D | ReshapeDynamic.cpp | 46 "shape_signature": [ in ReshapeDynamicFixture1()
77 "shape_signature": [ in ReshapeDynamicFixture1()
211 "shape_signature": [ in ReshapeDynamicFixture2()
241 "shape_signature": [ in ReshapeDynamicFixture2()
374 "shape_signature": [ in ReshapeDynamicFixture3()
405 "shape_signature": [ in ReshapeDynamicFixture3()
|
| D | FloorDiv.cpp | 44 "shape_signature": )" + inputShapeSignature1 + R"(, in FloorDivFixture()
56 "shape_signature": )" + inputShapeSignature2 + R"(, in FloorDivFixture()
68 "shape_signature": )" + outputShapeSignature + R"(, in FloorDivFixture()
|
| D | GatherNd.cpp | 40 "shape_signature": )" + paramsShape + R"( in GatherNdFixture()
67 "shape_signature": )" + outputShape + R"( in GatherNdFixture()
|
| D | ParserFlatbuffersFixture.hpp | 104 if (tensor->shape_signature.size() != 0) in MakeModelDynamic()
111 tensor->shape_signature.push_back(-1); in MakeModelDynamic()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | model_utils.cc | 86 const std::vector<int32_t>& shape_signature, in MakeTensor() argument
91 if (!shape_signature.empty()) { in MakeTensor()
92 tensor_raw->shape_signature = shape_signature; in MakeTensor()
102 const std::vector<int32_t>& shape_signature, in MakeTensorWithQuantParam() argument
106 MakeTensor(name, shape, shape_signature, type, tensor); in MakeTensorWithQuantParam()
|
| D | model_utils.h | 39 const std::vector<int32_t>& shape_signature,
45 const std::vector<int32_t>& shape_signature,
|
| D | quantize_model.cc | 428 utils::MakeTensor(leading_op_name, tensor->shape, tensor->shape_signature, in SetInputType()
444 leading_op_name, tensor->shape, tensor->shape_signature, input_type, in SetInputType()
490 utils::MakeTensor(tailing_op_name, tensor->shape, tensor->shape_signature, in SetOutputType()
506 tailing_op_name, tensor->shape, tensor->shape_signature, output_type, in SetOutputType()
767 input_tensor->shape_signature, activations_type, output_scale, in ApplyConstraints()
1019 tensor->shape_signature, activations_type, in QuantizeOpInput()
1049 tensor->shape_signature, TensorType_FLOAT32, &op_output); in QuantizeOpInput()
|
| D | modify_model_interface.cc | 417 added_tensor_name, tensor->shape, tensor->shape_signature, in AddUint8Dequant()
457 added_tensor_name, tensor->shape, tensor->shape_signature, in AddUint8Quant()
|
| D | quantize_weights_portable.cc | 469 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature,
553 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature, in QuantizeWeightsFloat16()
|
| D | quantize_weights.cc | 483 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature,
567 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature, in QuantizeWeightsFloat16()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | signature_def_with_multiple_entry_points.mlir | 20 // CHECK-NEXT: shape_signature: [ -1 ],
29 // CHECK-NEXT: shape_signature: [ -1 ],
38 // CHECK-NEXT: shape_signature: [ -1 ],
47 // CHECK-NEXT: shape_signature: [ -1 ],
76 // CHECK-NEXT: shape_signature: [ -1 ],
85 // CHECK-NEXT: shape_signature: [ -1 ],
94 // CHECK-NEXT: shape_signature: [ -1 ],
|
| D | signature_def.mlir | 18 // CHECK-NEXT: shape_signature: [ -1, 384 ],
27 // CHECK-NEXT: shape_signature: [ -1, 384 ],
60 // CHECK-NEXT: shape_signature: [ -1, 5 ],
69 // CHECK-NEXT: shape_signature: [ -1, 5 ],
|
| D | signature_def_output_override.mlir | 18 // CHECK-NEXT: shape_signature: [ -1, 3 ],
27 // CHECK-NEXT: shape_signature: [ -1, 3 ],
52 // CHECK-NEXT: shape_signature: [ -1, 5 ],
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/model_modifier/ |
| D | grafter.cc | 226 std::vector<int32_t> shape_signature; in AdjustTensors() local
227 if (i->shape_signature()) { in AdjustTensors()
228 shape_signature.assign(i->shape_signature()->cbegin(), in AdjustTensors()
229 i->shape_signature()->cend()); in AdjustTensors()
244 shape_signature.empty() ? 0 : fbb_->CreateVector(shape_signature))); in AdjustTensors()
|
| D | validation_graph_builder.cc | 157 std::vector<int32_t> shape_signature; in Tensors() local
158 if (tensor->shape_signature()) { in Tensors()
159 shape_signature.assign(tensor->shape_signature()->cbegin(), in Tensors()
160 tensor->shape_signature()->cend()); in Tensors()
161 if (shape_signature.size() >= 2 && shape_signature[0] == 1 && in Tensors()
163 shape_signature[0] = batch_size; in Tensors()
185 shape_signature.empty() ? 0 : fbb_.CreateVector(shape_signature))); in Tensors()
|
| /external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
| D | grafter.cc | 227 std::vector<int32_t> shape_signature; in AdjustTensors() local
228 if (i->shape_signature()) { in AdjustTensors()
229 shape_signature.assign(i->shape_signature()->cbegin(), in AdjustTensors()
230 i->shape_signature()->cend()); in AdjustTensors()
245 shape_signature.empty() ? 0 : fbb_->CreateVector(shape_signature))); in AdjustTensors()
|
| D | embedder.cc | 244 std::vector<int32_t> shape_signature; in Tensors() local
245 if (tensor->shape_signature()) { in Tensors()
246 shape_signature.assign(tensor->shape_signature()->cbegin(), in Tensors()
247 tensor->shape_signature()->cend()); in Tensors()
248 if (shape_signature.size() >= 2 && shape_signature[0] == 1 && in Tensors()
250 shape_signature[0] = batch_size; in Tensors()
272 shape_signature.empty() ? 0 : fbb_.CreateVector(shape_signature))); in Tensors()
|
| /external/tensorflow/tensorflow/lite/python/analyzer_wrapper/ |
| D | model_analyzer.cc | 109 if (tensor->shape_signature()) { in dump_tensor_detail()
111 for (int i = 0; i < tensor->shape_signature()->Length(); ++i) { in dump_tensor_detail()
112 const int j = tensor->shape_signature()->Get(i); in dump_tensor_detail()
114 if (i != tensor->shape_signature()->Length() - 1) { in dump_tensor_detail()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_v2_test_util.py | 54 for idx, (shape_signature, final_shape) in enumerate(input_shapes):
56 (input_details[idx]['shape_signature'] == shape_signature).all())
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.h | 117 std::vector<int> shape_signature = {})
134 shape_signature(shape_signature) {} in type()
149 std::vector<int> shape_signature; member
744 /*sparsity=*/0, builder_.CreateVector<int>(t.shape_signature)));
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib.cc | 257 std::vector<int32_t> shape_signature(shape_signature_view.begin(), in ExportTensors() local
259 shape_signature_offset = ExportVector<int32_t>(fbb, shape_signature); in ExportTensors()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_builder.cc | 626 if (tensor->shape_signature()) { in ParseTensors()
627 dims_signature = FlatBufferIntArrayToVector(tensor->shape_signature()); in ParseTensors()
|
| /external/tensorflow/tensorflow/lite/tools/versioning/ |
| D | op_signature.cc | 73 subgraph->tensors()->Get(tensor_no)->shape_signature(); in GetOpSignatureTensorSpecs()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 251 if (!tensor.shape_signature.empty()) { in GetTensorType()
252 llvm::SmallVector<int64_t, 4> shape(tensor.shape_signature.begin(), in GetTensorType()
253 tensor.shape_signature.end()); in GetTensorType()
|
| D | flatbuffer_export.cc | 894 std::vector<int32_t> shape_signature; in BuildTensor() local
919 shape_signature = std::vector<int32_t>(shape_ref.begin(), shape_ref.end()); in BuildTensor()
980 if (shape_signature.empty()) { in BuildTensor()
992 /*shape_signature=*/builder_.CreateVector(shape_signature), in BuildTensor()
|