Searched refs:sparsity (Results 1 – 8 of 8) sorted by relevance
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/ |
| A D | common.c | 112 if (sparsity == NULL) { in TfLiteSparsityFree()
116 if (sparsity->traversal_order) { in TfLiteSparsityFree()
118 sparsity->traversal_order = NULL; in TfLiteSparsityFree()
121 if (sparsity->block_map) { in TfLiteSparsityFree()
123 sparsity->block_map = NULL; in TfLiteSparsityFree()
126 if (sparsity->dim_metadata) { in TfLiteSparsityFree()
137 free(sparsity->dim_metadata); in TfLiteSparsityFree()
138 sparsity->dim_metadata = NULL; in TfLiteSparsityFree()
141 free(sparsity); in TfLiteSparsityFree()
155 TfLiteSparsityFree(t->sparsity); in TfLiteTensorFree()
[all …]
|
| A D | common.h | 449 TfLiteSparsity* sparsity; member
605 void TfLiteSparsityFree(TfLiteSparsity* sparsity);
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| A D | fully_connected.h | 31 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight() argument
48 const int w0_size = sparsity.dim_metadata[0].dense_size; in FullyConnectedSparseWeight()
49 const int* w1_segments = sparsity.dim_metadata[1].array_segments->data; in FullyConnectedSparseWeight()
50 const int* w1_indices = sparsity.dim_metadata[1].array_indices->data; in FullyConnectedSparseWeight()
77 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight1x4Impl() argument
96 const int* w1_segments = sparsity.dim_metadata[1].array_segments->data; in FullyConnectedSparseWeight1x4Impl()
97 const int* w1_indices = sparsity.dim_metadata[1].array_indices->data; in FullyConnectedSparseWeight1x4Impl()
117 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight1x4Task()
123 : sparsity(sparsity), in FullyConnectedSparseWeight1x4Task()
145 const TfLiteSparsity& sparsity;
[all …]
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/ |
| A D | densify.h | 29 inline void Densify(const TfLiteSparsity* sparsity, in Densify() argument
39 tflite::optimize::sparsity::FormatConverter<T> converter(vector_shape, in Densify()
40 *sparsity); in Densify()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/sparse_ops/ |
| A D | fully_connected.h | 26 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight() argument
35 tflite::optimize::sparsity::FormatConverter<float> converter( in FullyConnectedSparseWeight()
36 weights_shape_vector, sparsity); in FullyConnectedSparseWeight()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/ |
| A D | interpreter_builder.h | 116 TfLiteSparsity** sparsity);
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/ |
| A D | test_util.h | 224 tflite::optimize::sparsity::FormatConverter<int8_t> converter( in AddConstSparseInput()
296 tflite::optimize::sparsity::FormatConverter<T> converter(
578 if (tensor->sparsity) { in ExtractVector()
582 tensor_size = tensor->sparsity in ExtractVector()
583 ->dim_metadata[tensor->sparsity->dim_metadata_size - 1] in ExtractVector()
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/schema/ |
| A D | schema_generated.h | 3788 std::unique_ptr<tflite::SparsityParametersT> sparsity;
3827 const tflite::SparsityParameters *sparsity() const {
3845 verifier.VerifyTable(sparsity()) &&
3876 void add_sparsity(flatbuffers::Offset<tflite::SparsityParameters> sparsity) {
3877 fbb_.AddOffset(Tensor::VT_SPARSITY, sparsity);
3902 flatbuffers::Offset<tflite::SparsityParameters> sparsity = 0,
3906 builder_.add_sparsity(sparsity);
3924 flatbuffers::Offset<tflite::SparsityParameters> sparsity = 0,
3937 sparsity,
12036 …{ auto _e = sparsity(); if (_e) _o->sparsity = std::unique_ptr<tflite::SparsityParametersT>(_e->Un…
[all …]
|
Completed in 35 milliseconds