| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | model_utils.cc | 85 const std::vector<int32_t>& shape_signature, in MakeTensor() argument
90 if (!shape_signature.empty()) { in MakeTensor()
91 tensor_raw->shape_signature = shape_signature; in MakeTensor()
101 const std::vector<int32_t>& shape_signature, in MakeTensorWithQuantParam() argument
105 MakeTensor(name, shape, shape_signature, type, tensor); in MakeTensorWithQuantParam()
|
| D | model_utils.h | 37 const std::vector<int32_t>& shape_signature,
43 const std::vector<int32_t>& shape_signature,
|
| D | quantize_weights.cc | 328 const std::vector<int32_t>& shape_signature, in MakeTensor() argument
333 if (!shape_signature.empty()) { in MakeTensor()
334 tensor_raw->shape_signature = shape_signature; in MakeTensor()
509 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature, in QuantizeWeightsInt8()
593 utils::MakeTensor(dequant_name, tensor->shape, tensor->shape_signature, in QuantizeWeightsFloat16()
|
| D | quantize_model.cc | 277 utils::MakeTensor(leading_op_name, tensor->shape, tensor->shape_signature, in SetInputType()
293 leading_op_name, tensor->shape, tensor->shape_signature, input_type, in SetInputType()
339 utils::MakeTensor(tailing_op_name, tensor->shape, tensor->shape_signature, in SetOutputType()
355 tailing_op_name, tensor->shape, tensor->shape_signature, output_type, in SetOutputType()
491 input_tensor->shape_signature, activations_type, output_scale, in ApplyConstraints()
740 tensor->shape_signature, activations_type, in QuantizeOpInput()
770 tensor->shape_signature, TensorType_FLOAT32, &op_output); in QuantizeOpInput()
|
| D | modify_model_interface.cc | 414 added_tensor_name, tensor->shape, tensor->shape_signature, in AddUint8Dequant()
454 added_tensor_name, tensor->shape, tensor->shape_signature, in AddUint8Quant()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/ |
| D | signature_def.mlir | 18 // CHECK-NEXT: shape_signature: [ -1, 384 ]
26 // CHECK-NEXT: shape_signature: [ -1, 384 ]
55 // CHECK-NEXT: shape_signature: [ -1, 5 ]
63 // CHECK-NEXT: shape_signature: [ -1, 5 ]
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | lite_v2_test_util.py | 53 for idx, (shape_signature, final_shape) in enumerate(input_shapes):
55 (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
687 /*sparsity=*/0, builder_.CreateVector<int>(t.shape_signature)));
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_builder.cc | 598 if (tensor->shape_signature()) { in ParseTensors()
599 dims_signature = FlatBufferIntArrayToVector(tensor->shape_signature()); in ParseTensors()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 218 if (!tensor.shape_signature.empty()) { in GetTensorType()
219 llvm::SmallVector<int64_t, 4> shape(tensor.shape_signature.begin(), in GetTensorType()
220 tensor.shape_signature.end()); in GetTensorType()
|
| D | flatbuffer_export.cc | 760 std::vector<int32_t> shape_signature; in BuildTensor() local
785 shape_signature = std::vector<int32_t>(shape_ref.begin(), shape_ref.end()); in BuildTensor()
836 if (shape_signature.empty()) { in BuildTensor()
846 /*shape_signature=*/builder_.CreateVector(shape_signature)); in BuildTensor()
|
| /external/tensorflow/tensorflow/lite/schema/ |
| D | schema_generated.h | 3702 std::vector<int32_t> shape_signature;
3743 const flatbuffers::Vector<int32_t> *shape_signature() const {
3760 verifier.VerifyVector(shape_signature()) &&
3792 void add_shape_signature(flatbuffers::Offset<flatbuffers::Vector<int32_t>> shape_signature) {
3793 fbb_.AddOffset(Tensor::VT_SHAPE_SIGNATURE, shape_signature);
3816 flatbuffers::Offset<flatbuffers::Vector<int32_t>> shape_signature = 0) {
3818 builder_.add_shape_signature(shape_signature);
3838 const std::vector<int32_t> *shape_signature = nullptr) {
3841 auto shape_signature__ = shape_signature ? _fbb.CreateVector<int32_t>(*shape_signature) : 0;
11712 …= shape_signature(); if (_e) { _o->shape_signature.resize(_e->size()); for (flatbuffers::uoffset_t…
[all …]
|
| D | schema_v3a.fbs | 212 shape_signature:[int]; // Optional.
|
| D | schema.fbs | 216 shape_signature:[int]; // Optional.
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | test_schema.fbs | 209 shape_signature:[int]; // Optional.
|