Searched refs:traversal_order (Results 1 – 8 of 8) sorted by relevance
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/third_party/ruy/ruy/ |
| A D | block_map_test.cc | 150 BlockMapTraversalOrder traversal_order) { in GetBlockByIndexSquareTest() argument
157 block_map.traversal_order = traversal_order; in GetBlockByIndexSquareTest()
202 switch (traversal_order) { in GetBlockByIndexSquareTest()
244 for (BlockMapTraversalOrder traversal_order : in TEST()
248 GetBlockByIndexSquareTest(num_blocks_base_log2, traversal_order); in TEST()
|
| A D | block_map.h | 87 BlockMapTraversalOrder traversal_order; member
|
| A D | block_map.cc | 105 switch (block_map.traversal_order) { in GetBlockByIndex()
116 RUY_DCHECK(block_map.traversal_order == BlockMapTraversalOrder::kLinear); in GetBlockByIndex()
464 block_map->traversal_order = GetTraversalOrder( in MakeBlockMap()
|
| A D | trace.h | 691 block_map->traversal_order == BlockMapTraversalOrder::kLinear ? "linear" \
692 : block_map->traversal_order == BlockMapTraversalOrder::kFractalZ \
694 : block_map->traversal_order == BlockMapTraversalOrder::kFractalU \
696 : block_map->traversal_order == BlockMapTraversalOrder::kFractalHilbert \
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/ |
| A D | common.c | 116 if (sparsity->traversal_order) { in TfLiteSparsityFree()
117 TfLiteIntArrayFree(sparsity->traversal_order); in TfLiteSparsityFree()
118 sparsity->traversal_order = NULL; in TfLiteSparsityFree()
|
| A D | common.h | 364 TfLiteIntArray* traversal_order; member
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/ |
| A D | test_util.h | 114 int32_t channel_index = 0, std::vector<int> traversal_order = {},
130 traversal_order(traversal_order), in type()
145 std::vector<int> traversal_order; member
221 const int dims_count = t.traversal_order.size(); in AddConstSparseInput()
225 t.shape, t.traversal_order, t.format, t.block_size, t.block_map); in AddConstSparseInput()
237 t.format[t.traversal_order[i]] == kTfLiteDimSparseCSR) { in AddConstSparseInput()
257 builder_, builder_.CreateVector(t.traversal_order), in AddConstSparseInput()
293 const int dims_count = t.traversal_order.size();
297 t.shape, t.traversal_order, t.format, t.block_size, t.block_map);
309 t.format[t.traversal_order[i]] == kTfLiteDimSparseCSR) {
[all …]
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/schema/ |
| A D | schema_generated.h | 3688 std::vector<int32_t> traversal_order;
3702 const flatbuffers::Vector<int32_t> *traversal_order() const {
3714 verifier.VerifyVector(traversal_order()) &&
3730 void add_traversal_order(flatbuffers::Offset<flatbuffers::Vector<int32_t>> traversal_order) {
3731 fbb_.AddOffset(SparsityParameters::VT_TRAVERSAL_ORDER, traversal_order);
3753 flatbuffers::Offset<flatbuffers::Vector<int32_t>> traversal_order = 0,
3759 builder_.add_traversal_order(traversal_order);
3765 const std::vector<int32_t> *traversal_order = nullptr,
3768 auto traversal_order__ = traversal_order ? _fbb.CreateVector<int32_t>(*traversal_order) : 0;
11998 …= traversal_order(); if (_e) { _o->traversal_order.resize(_e->size()); for (flatbuffers::uoffset_t…
[all …]
|
Completed in 30 milliseconds