Home
last modified time | relevance | path

Searched refs:dim_metadata (Results 1 – 5 of 5) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/sparse_ops/
A Dfully_connected.h48 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()
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()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dcommon.c126 if (sparsity->dim_metadata) { in TfLiteSparsityFree()
129 TfLiteDimensionMetadata metadata = sparsity->dim_metadata[i]; in TfLiteSparsityFree()
137 free(sparsity->dim_metadata); in TfLiteSparsityFree()
138 sparsity->dim_metadata = NULL; in TfLiteSparsityFree()
A Dcommon.h366 TfLiteDimensionMetadata* dim_metadata; member
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/
A Dtest_util.h228 const auto& dim_metadata = converter.GetDimMetadata(); in AddConstSparseInput() local
240 builder_.CreateVector(dim_metadata[metadata_idx])) in AddConstSparseInput()
244 builder_, builder_.CreateVector(dim_metadata[metadata_idx + 1])) in AddConstSparseInput()
252 builder_, DimensionType_DENSE, dim_metadata[metadata_idx][0]); in AddConstSparseInput()
300 const auto dim_metadata = converter.GetDimMetadata();
312 builder_.CreateVector(dim_metadata[metadata_idx]))
316 builder_, builder_.CreateVector(dim_metadata[metadata_idx + 1]))
324 builder_, DimensionType_DENSE, dim_metadata[metadata_idx][0]);
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 Dschema_generated.h3690 std::vector<std::unique_ptr<tflite::DimensionMetadataT>> dim_metadata;
3708 const flatbuffers::Vector<flatbuffers::Offset<tflite::DimensionMetadata>> *dim_metadata() const {
3718 verifier.VerifyVector(dim_metadata()) &&
3719 verifier.VerifyVectorOfTables(dim_metadata()) &&
3737 fbb_.AddOffset(SparsityParameters::VT_DIM_METADATA, dim_metadata);
3755 …rs::Offset<flatbuffers::Vector<flatbuffers::Offset<tflite::DimensionMetadata>>> dim_metadata = 0) {
3757 builder_.add_dim_metadata(dim_metadata);
3767 const std::vector<flatbuffers::Offset<tflite::DimensionMetadata>> *dim_metadata = nullptr) {
3770 …auto dim_metadata__ = dim_metadata ? _fbb.CreateVector<flatbuffers::Offset<tflite::DimensionMetada…
12000 … _e = dim_metadata(); if (_e) { _o->dim_metadata.resize(_e->size()); for (flatbuffers::uoffset_t _…
[all …]

Completed in 27 milliseconds