Home
last modified time | relevance | path

Searched refs:traversal_order (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/
Dformat_converter_test.cc28 const std::vector<int> traversal_order = {0, 1}; in TEST() local
31 FormatConverter<int> converter(dense_shape, traversal_order, format); in TEST()
57 const std::vector<int> traversal_order = {0, 1}; in TEST() local
60 FormatConverter<int> converter(dense_shape, traversal_order, format); in TEST()
88 const std::vector<int> traversal_order = {0, 1}; in TEST() local
91 FormatConverter<int> converter(dense_shape, traversal_order, format); in TEST()
119 const std::vector<int> traversal_order = {0, 1}; in TEST() local
122 FormatConverter<int> converter(dense_shape, traversal_order, format); in TEST()
152 const std::vector<int> traversal_order = {1, 0}; in TEST() local
155 FormatConverter<int> converter(dense_shape, traversal_order, format); in TEST()
[all …]
Dformat_converter.cc54 const std::vector<int>& shape, const std::vector<int>& traversal_order, in FormatConverter() argument
58 traversal_order_(traversal_order), in FormatConverter()
66 format_[i] = format[traversal_order[i]]; in FormatConverter()
224 traversal_order_ = TfLiteIntArrayToVector(sparsity.traversal_order); in FormatConverter()
Dformat_converter.h44 const std::vector<int>& traversal_order,
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddense_to_sparse.cc51 std::vector<int>* traversal_order, in PopulateEncodingParams() argument
55 traversal_order->resize(dims_count); in PopulateEncodingParams()
58 (*traversal_order)[i] = i; in PopulateEncodingParams()
69 traversal_order->push_back(block_rank + dims_count); in PopulateEncodingParams()
111 std::vector<int> traversal_order = {}; in CalculateBlockSparsity() local
115 PopulateEncodingParams(block_size, &traversal_order, &format, &b_map, in CalculateBlockSparsity()
120 shape, traversal_order, format, b_size, b_map); in CalculateBlockSparsity()
130 shape, traversal_order, format, b_size, b_map); in CalculateBlockSparsity()
141 shape, traversal_order, format, b_size, b_map); in CalculateBlockSparsity()
237 std::vector<int> traversal_order = {}; in BuildSparsityParameterAttribute() local
[all …]
/external/ruy/ruy/
Dblock_map_test.cc150 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()
Dblock_map.h87 BlockMapTraversalOrder traversal_order; member
Dblock_map.cc104 switch (block_map.traversal_order) { in GetBlockByIndex()
115 RUY_DCHECK(block_map.traversal_order == BlockMapTraversalOrder::kLinear); in GetBlockByIndex()
463 block_map->traversal_order = GetTraversalOrder( in MakeBlockMap()
Dtrace.h691 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 \
/external/tensorflow/tensorflow/lite/tools/
Dverifier.cc238 const int total_level = sparsity.traversal_order()->size(); in VerifyAndCountElements()
241 const int original_dim = sparsity.traversal_order()->Get(i); in VerifyAndCountElements()
294 if (sparsity->traversal_order() == nullptr || in VerifyAndCountSparseElements()
299 const int total_dims = sparsity->traversal_order()->size(); in VerifyAndCountSparseElements()
321 std::vector<int> traversal_order(total_dims); in VerifyAndCountSparseElements() local
323 traversal_order[i] = sparsity->traversal_order()->Get(i); in VerifyAndCountSparseElements()
326 std::sort(traversal_order.begin(), traversal_order.begin() + original_rank); in VerifyAndCountSparseElements()
328 if (traversal_order[i] != i) { in VerifyAndCountSparseElements()
333 std::sort(traversal_order.begin() + original_rank, traversal_order.end()); in VerifyAndCountSparseElements()
335 if (traversal_order[i] != i) { in VerifyAndCountSparseElements()
[all …]
/external/tensorflow/tensorflow/lite/kernels/
Ddensify_test.cc72 input.traversal_order = {0, 1}; in TEST()
86 input.traversal_order = {0, 1, 2}; in TEST()
100 input.traversal_order = {0, 1}; in TEST()
Dtest_util.h114 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 …]
Dfully_connected_test.cc1253 weight.traversal_order = {0, 1}; in TEST_P()
1278 weight.traversal_order = {0, 1}; in TEST_P()
1305 weight.traversal_order = {0, 1, 2}; in TEST_P()
1335 weight.traversal_order = {0, 1, 2}; in TEST_P()
1368 weight.traversal_order = {0, 1, 2}; in TEST_P()
1427 weight.traversal_order = {0, 1, 2}; in TEST_P()
1483 weight.traversal_order = {0, 1, 2}; in TEST_P()
/external/tensorflow/tensorflow/lite/
Dmodel_test.cc424 TfLiteIntArray* traversal_order = TfLiteIntArrayCreate(4); in TEST() local
425 traversal_order->data[0] = 0; in TEST()
426 traversal_order->data[1] = 1; in TEST()
427 traversal_order->data[2] = 2; in TEST()
428 traversal_order->data[3] = 3; in TEST()
430 TfLiteIntArrayEqual(t1->sparsity->traversal_order, traversal_order)); in TEST()
431 TfLiteIntArrayFree(traversal_order); in TEST()
Dinterpreter_builder.cc428 if (src_sparsity->traversal_order() == nullptr || in ParseSparsity()
439 const size_t traversal_order_size = src_sparsity->traversal_order()->size(); in ParseSparsity()
440 sparsity->traversal_order = TfLiteIntArrayCreate(traversal_order_size); in ParseSparsity()
442 sparsity->traversal_order->data[i] = in ParseSparsity()
443 src_sparsity->traversal_order()->Get(i); in ParseSparsity()
/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/python/
Dformat_converter_extension_test.py39 traversal_order = [0, 1, 2, 3]
47 traversal_order, dim_types,
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dbinary_elementwise_tester.cc269 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
271 traversal_order[i] = i; in CreateTfLiteModel()
276 CreateSparsityParameters(builder, builder.CreateVector(traversal_order), in CreateTfLiteModel()
295 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
297 traversal_order[i] = i; in CreateTfLiteModel()
302 CreateSparsityParameters(builder, builder.CreateVector(traversal_order), in CreateTfLiteModel()
Dprelu_tester.cc151 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
153 traversal_order[i] = i; in CreateTfLiteModel()
158 CreateSparsityParameters(builder, builder.CreateVector(traversal_order), in CreateTfLiteModel()
Dconv_2d_tester.cc259 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
261 traversal_order[i] = i; in CreateTfLiteModel()
266 CreateSparsityParameters(builder, builder.CreateVector(traversal_order), in CreateTfLiteModel()
Ddepthwise_conv_2d_tester.cc261 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
263 traversal_order[i] = i; in CreateTfLiteModel()
268 CreateSparsityParameters(builder, builder.CreateVector(traversal_order), in CreateTfLiteModel()
/external/tensorflow/tensorflow/lite/c/
Dcommon.c114 if (sparsity->traversal_order) { in TfLiteSparsityFree()
115 TfLiteIntArrayFree(sparsity->traversal_order); in TfLiteSparsityFree()
116 sparsity->traversal_order = NULL; in TfLiteSparsityFree()
Dcommon_test.cc124 t.sparsity->traversal_order = TfLiteIntArrayCreate(2); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_structs.td36 StructFieldAttr<"traversal_order", I32ArrayAttr>,
/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/
Dtest_schema.fbs160 // - if not block sparse, the traversal_order is just a permutation of (d0,
162 // have traversal_order = (d0, d1).
164 // traversal_order has n + k elements. The first n elements are still a
168 // would have traversal_order = (d0, d1, d2, d3).
169 traversal_order:[int];
178 // The size of the dim_metadata array = the size of the traversal_order array
/external/tensorflow/tensorflow/lite/schema/
Dschema_v3a.fbs163 // - if not block sparse, the traversal_order is just a permutation of (d0,
165 // have traversal_order = (d0, d1).
167 // traversal_order has n + k elements. The first n elements are still a
171 // would have traversal_order = (d0, d1, d2, d3).
172 traversal_order:[int];
181 // The size of the dim_metadata array = the size of the traversal_order array
Dschema.fbs167 // - if not block sparse, the traversal_order is just a permutation of (d0,
169 // have traversal_order = (d0, d1).
171 // traversal_order has n + k elements. The first n elements are still a
175 // would have traversal_order = (d0, d1, d2, d3).
176 traversal_order:[int];
185 // The size of the dim_metadata array = the size of the traversal_order array

12