Home
last modified time | relevance | path

Searched refs:block_map (Results 1 – 11 of 11) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/third_party/ruy/ruy/
A Dblock_map.h121 const CpuCacheParams& cpu_cache_params, BlockMap* block_map);
124 void GetBlockByIndex(const BlockMap& block_map, int index,
131 void GetBlockMatrixCoords(Side side, const BlockMap& block_map, int block,
137 void GetBlockMatrixCoords(const BlockMap& block_map, const SidePair<int>& block,
142 inline int NumBlocksPerSide(Side side, const BlockMap& block_map) { in NumBlocksPerSide() argument
143 return 1 << (block_map.num_blocks_base_log2 + in NumBlocksPerSide()
144 block_map.rectangularness_log2[side]); in NumBlocksPerSide()
154 inline int NumBlocks(const BlockMap& block_map) { in NumBlocks() argument
155 return 1 << (2 * block_map.num_blocks_base_log2 + in NumBlocks()
156 block_map.rectangularness_log2[Side::kLhs] + in NumBlocks()
[all …]
A Dblock_map_test.cc44 BlockMap block_map; in MakeBlockMapTuningTest() local
47 &block_map); in MakeBlockMapTuningTest()
155 BlockMap block_map; in GetBlockByIndexSquareTest() local
156 block_map.thread_count = 1; in GetBlockByIndexSquareTest()
157 block_map.traversal_order = traversal_order; in GetBlockByIndexSquareTest()
160 block_map.dims[side] = 1 << size_log2; in GetBlockByIndexSquareTest()
161 block_map.rectangularness_log2[side] = 0; in GetBlockByIndexSquareTest()
162 block_map.kernel_dims[side] = 1 << kKernelSizeLog2; in GetBlockByIndexSquareTest()
163 block_map.small_block_dims[side] = block_map.kernel_dims[side]; in GetBlockByIndexSquareTest()
164 block_map.large_blocks[side] = 0; in GetBlockByIndexSquareTest()
[all …]
A Dblock_map.cc99 1u << (2 * block_map.num_blocks_base_log2); in GetBlockByIndex()
105 switch (block_map.traversal_order) { in GetBlockByIndex()
451 block_map->dims[Side::kLhs] = rows; in MakeBlockMap()
452 block_map->dims[Side::kRhs] = cols; in MakeBlockMap()
453 block_map->kernel_dims[Side::kLhs] = kernel_rows; in MakeBlockMap()
464 block_map->traversal_order = GetTraversalOrder( in MakeBlockMap()
468 block_map->thread_count = in MakeBlockMap()
478 block_map.kernel_dims[side]; in GetBlockMatrixCoords()
480 *start + block_map.small_block_dims[side] + in GetBlockMatrixCoords()
481 (block < block_map.large_blocks[side] ? block_map.kernel_dims[side] : 0); in GetBlockMatrixCoords()
[all …]
A Dtrace.h669 block_map->small_block_dims[Side::kLhs], \
670 block_map->small_block_dims[Side::kRhs], \
671 block_map->small_block_dims[Side::kLhs] + \
672 block_map->kernel_dims[Side::kLhs], \
673 block_map->small_block_dims[Side::kRhs] + \
674 block_map->kernel_dims[Side::kRhs]); \
678 block_map->large_blocks[Side::kLhs], \
679 block_map->small_block_dims[Side::kLhs] + \
680 block_map->kernel_dims[Side::kLhs], \
681 block_map->small_block_dims[Side::kLhs]); \
[all …]
A Dtrmul.cc59 TrMulTask(TrMulParams* params, const BlockMap& block_map, in TrMulTask() argument
65 block_map_(block_map), in TrMulTask()
345 BlockMap block_map; in TrMul() local
349 tentative_thread_count, cpu_cache_params, &block_map); in TrMul()
352 const int thread_count = block_map.thread_count; in TrMul()
365 const int size = NumBlocksPerSide(side, block_map); in TrMul()
390 new (tasks + i) TrMulTask(params, block_map, atomic_block_id, i, in TrMul()
A DBUILD296 name = "block_map",
298 "block_map.cc",
301 "block_map.h",
320 ":block_map",
963 ":block_map",
A DCMakeLists.txt308 block_map.cc
310 block_map.h
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/c/
A Dcommon.c121 if (sparsity->block_map) { in TfLiteSparsityFree()
122 TfLiteIntArrayFree(sparsity->block_map); in TfLiteSparsityFree()
123 sparsity->block_map = NULL; in TfLiteSparsityFree()
A Dcommon.h365 TfLiteIntArray* block_map; member
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/
A Dtest_util.h116 std::vector<int> block_size = {}, std::vector<int> block_map = {},
133 block_map(block_map), in type()
148 std::vector<int> block_map; member
225 t.shape, t.traversal_order, t.format, t.block_size, t.block_map); in AddConstSparseInput()
258 builder_.CreateVector(t.block_map), in AddConstSparseInput()
297 t.shape, t.traversal_order, t.format, t.block_size, t.block_map);
330 builder_.CreateVector(t.block_map),
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/schema/
A Dschema_generated.h3689 std::vector<int32_t> block_map;
3705 const flatbuffers::Vector<int32_t> *block_map() const {
3716 verifier.VerifyVector(block_map()) &&
3733 void add_block_map(flatbuffers::Offset<flatbuffers::Vector<int32_t>> block_map) {
3734 fbb_.AddOffset(SparsityParameters::VT_BLOCK_MAP, block_map);
3754 flatbuffers::Offset<flatbuffers::Vector<int32_t>> block_map = 0,
3758 builder_.add_block_map(block_map);
3766 const std::vector<int32_t> *block_map = nullptr,
3769 auto block_map__ = block_map ? _fbb.CreateVector<int32_t>(*block_map) : 0;
11999 … auto _e = block_map(); if (_e) { _o->block_map.resize(_e->size()); for (flatbuffers::uoffset_t _i…
[all …]

Completed in 33 milliseconds