| /external/XNNPACK/test/ |
| D | f32-spmm-minmax.cc | 29 .sparsity(0.0f) in TEST()
42 .sparsity(0.0f) in TEST()
57 .sparsity(0.0f) in TEST()
74 .sparsity(0.0f) in TEST()
92 .sparsity(0.0f) in TEST()
110 .sparsity(0.0f) in TEST()
128 .sparsity(0.0f) in TEST()
144 .sparsity(0.0f) in TEST()
161 .sparsity(0.0f) in TEST()
178 .sparsity(0.5f) in TEST()
[all …]
|
| D | f16-spmm-minmax.cc | 29 .sparsity(0.0f) in TEST()
42 .sparsity(0.0f) in TEST()
57 .sparsity(0.0f) in TEST()
74 .sparsity(0.0f) in TEST()
92 .sparsity(0.0f) in TEST()
110 .sparsity(0.0f) in TEST()
128 .sparsity(0.0f) in TEST()
144 .sparsity(0.0f) in TEST()
161 .sparsity(0.0f) in TEST()
178 .sparsity(0.5f) in TEST()
[all …]
|
| D | convolution-nchw.cc | 20 .sparsity(0.5f)
32 .sparsity(1.0f)
45 .sparsity(0.5f)
59 .sparsity(0.5f)
73 .sparsity(0.5f)
87 .sparsity(0.5f)
99 .sparsity(0.5f)
111 .sparsity(0.5f)
124 .sparsity(0.5f)
139 .sparsity(0.5f) in TEST()
[all …]
|
| D | spmm-microkernel-tester.h | 99 inline SpMMMicrokernelTester& sparsity(float sparsity) { in sparsity() function
100 this->sparsity_ = sparsity; in sparsity()
104 inline float sparsity() const { in sparsity() function
169 if (prng() <= sparsity()) {
342 if (prng() <= sparsity()) { in Test()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.c | 109 void TfLiteSparsityFree(TfLiteSparsity* sparsity) { in TfLiteSparsityFree() argument
110 if (sparsity == NULL) { in TfLiteSparsityFree()
114 if (sparsity->traversal_order) { in TfLiteSparsityFree()
115 TfLiteIntArrayFree(sparsity->traversal_order); in TfLiteSparsityFree()
116 sparsity->traversal_order = NULL; in TfLiteSparsityFree()
119 if (sparsity->block_map) { in TfLiteSparsityFree()
120 TfLiteIntArrayFree(sparsity->block_map); in TfLiteSparsityFree()
121 sparsity->block_map = NULL; in TfLiteSparsityFree()
124 if (sparsity->dim_metadata) { in TfLiteSparsityFree()
126 for (; i < sparsity->dim_metadata_size; i++) { in TfLiteSparsityFree()
[all …]
|
| D | common_test.cc | 106 t.sparsity = nullptr; in TEST()
123 t.sparsity = static_cast<TfLiteSparsity*>(malloc(sizeof(TfLiteSparsity))); in TEST()
124 t.sparsity->traversal_order = TfLiteIntArrayCreate(2); in TEST()
125 t.sparsity->block_map = nullptr; in TEST()
127 t.sparsity->dim_metadata = static_cast<TfLiteDimensionMetadata*>( in TEST()
129 t.sparsity->dim_metadata_size = 2; in TEST()
131 t.sparsity->dim_metadata[0].format = kTfLiteDimDense; in TEST()
132 t.sparsity->dim_metadata[0].dense_size = 4; in TEST()
134 t.sparsity->dim_metadata[1].format = kTfLiteDimSparseCSR; in TEST()
135 t.sparsity->dim_metadata[1].array_segments = TfLiteIntArrayCreate(2); in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
| 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()
139 sparsity, params, input_shape, input_data, weights_shape, weights_data, in Run()
[all …]
|
| /external/tensorflow/tensorflow/lite/ |
| D | model_test.cc | 430 TfLiteIntArrayEqual(t1->sparsity->traversal_order, traversal_order)); in TEST()
436 ASSERT_TRUE(TfLiteIntArrayEqual(t1->sparsity->block_map, block_map)); in TEST()
439 ASSERT_EQ(t1->sparsity->dim_metadata_size, 4); in TEST()
441 ASSERT_EQ(t1->sparsity->dim_metadata[0].format, kTfLiteDimDense); in TEST()
442 ASSERT_EQ(t1->sparsity->dim_metadata[0].dense_size, 2); in TEST()
443 ASSERT_EQ(t1->sparsity->dim_metadata[0].array_segments, nullptr); in TEST()
444 ASSERT_EQ(t1->sparsity->dim_metadata[0].array_indices, nullptr); in TEST()
446 ASSERT_EQ(t1->sparsity->dim_metadata[1].format, kTfLiteDimSparseCSR); in TEST()
447 ASSERT_EQ(t1->sparsity->dim_metadata[1].dense_size, 0); in TEST()
452 ASSERT_TRUE(TfLiteIntArrayEqual(t1->sparsity->dim_metadata[1].array_segments, in TEST()
[all …]
|
| D | interpreter_builder.cc | 434 auto* sparsity = in ParseSparsity() local
436 memset(sparsity, 0, sizeof(TfLiteSparsity)); in ParseSparsity()
437 *sparsity_ptr = sparsity; in ParseSparsity()
440 sparsity->traversal_order = TfLiteIntArrayCreate(traversal_order_size); in ParseSparsity()
442 sparsity->traversal_order->data[i] = in ParseSparsity()
448 sparsity->block_map = TfLiteIntArrayCreate(block_map_size); in ParseSparsity()
450 sparsity->block_map->data[i] = src_sparsity->block_map()->Get(i); in ParseSparsity()
455 sparsity->dim_metadata_size = dim_metadata_size; in ParseSparsity()
456 sparsity->dim_metadata = reinterpret_cast<TfLiteDimensionMetadata*>( in ParseSparsity()
458 memset(sparsity->dim_metadata, 0, in ParseSparsity()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/ |
| D | verifier.cc | 237 const SparsityParameters& sparsity, const std::vector<int>& dim_sizes) { in VerifyAndCountElements() argument
238 const int total_level = sparsity.traversal_order()->size(); in VerifyAndCountElements()
241 const int original_dim = sparsity.traversal_order()->Get(i); in VerifyAndCountElements()
242 const auto* dim_metadata = sparsity.dim_metadata()->Get(i); in VerifyAndCountElements()
293 const auto* sparsity = tensor.sparsity(); in VerifyAndCountSparseElements() local
294 if (sparsity->traversal_order() == nullptr || in VerifyAndCountSparseElements()
295 sparsity->dim_metadata() == nullptr) { in VerifyAndCountSparseElements()
299 const int total_dims = sparsity->traversal_order()->size(); in VerifyAndCountSparseElements()
301 const int sparsity_dim_metadata_size = sparsity->dim_metadata()->size(); in VerifyAndCountSparseElements()
308 if (sparsity->block_map() == nullptr) { in VerifyAndCountSparseElements()
[all …]
|
| D | verifier_test.cc | 608 tensor->sparsity->block_map = {}; in TEST()
632 tensor->sparsity->dim_metadata[1]->array_indices.AsUint8Vector()->values[1] = in TEST()
683 tensor->sparsity->traversal_order[0] = 10; in TEST()
707 tensor->sparsity->traversal_order = {1, 0, 3, 2}; in TEST()
708 tensor->sparsity->block_map = {0, 1}; in TEST()
709 tensor->sparsity->dim_metadata[0]->format = DimensionType_DENSE; in TEST()
710 tensor->sparsity->dim_metadata[0]->dense_size = 2; in TEST()
712 tensor->sparsity->dim_metadata[1]->format = DimensionType_SPARSE_CSR; in TEST()
713 tensor->sparsity->dim_metadata[1]->array_segments.AsUint8Vector()->values = { in TEST()
715 tensor->sparsity->dim_metadata[1]->array_indices.AsUint8Vector()->values = { in TEST()
[all …]
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter.cc | 22 namespace sparsity { namespace
217 const TfLiteSparsity& sparsity) in FormatConverter() argument
224 traversal_order_ = TfLiteIntArrayToVector(sparsity.traversal_order); in FormatConverter()
225 block_map_ = TfLiteIntArrayToVector(sparsity.block_map); in FormatConverter()
227 format_.resize(sparsity.dim_metadata_size); in FormatConverter()
228 dim_metadata_.resize(2 * sparsity.dim_metadata_size); in FormatConverter()
229 for (int i = 0; i < sparsity.dim_metadata_size; i++) { in FormatConverter()
230 format_[i] = sparsity.dim_metadata[i].format; in FormatConverter()
232 dim_metadata_[2 * i] = {sparsity.dim_metadata[i].dense_size}; in FormatConverter()
235 TfLiteIntArrayToVector(sparsity.dim_metadata[i].array_segments); in FormatConverter()
[all …]
|
| D | format_converter.h | 25 namespace sparsity {
54 const TfLiteSparsity& sparsity);
|
| /external/XNNPACK/models/ |
| D | models.h | 23 ExecutionPlan FP32SparseMobileNetV1(float sparsity, pthreadpool_t threadpool);
24 ExecutionPlan FP32SparseMobileNetV2(float sparsity, pthreadpool_t threadpool);
25 ExecutionPlan FP32SparseMobileNetV3Large(float sparsity, pthreadpool_t threadpool);
26 ExecutionPlan FP32SparseMobileNetV3Small(float sparsity, pthreadpool_t threadpool);
|
| D | fp32-sparse-mobilenet-v2.cc | 19 ExecutionPlan FP32SparseMobileNetV2(float sparsity, pthreadpool_t threadpool) { in FP32SparseMobileNetV2() argument
265 std::generate(w69.begin(), w69.end() - size_t(sparsity * w69.size()), std::ref(f32rng)); in FP32SparseMobileNetV2()
269 std::generate(w71.begin(), w71.end() - size_t(sparsity * w71.size()), std::ref(f32rng)); in FP32SparseMobileNetV2()
275 std::generate(w75.begin(), w75.end() - size_t(sparsity * w75.size()), std::ref(f32rng)); in FP32SparseMobileNetV2()
279 std::generate(w77.begin(), w77.end() - size_t(sparsity * w77.size()), std::ref(f32rng)); in FP32SparseMobileNetV2()
285 std::generate(w81.begin(), w81.end() - size_t(sparsity * w81.size()), std::ref(f32rng)); in FP32SparseMobileNetV2()
289 std::generate(w83.begin(), w83.end() - size_t(sparsity * w83.size()), std::ref(f32rng)); in FP32SparseMobileNetV2()
295 std::generate(w87.begin(), w87.end() - size_t(sparsity * w87.size()), std::ref(f32rng)); in FP32SparseMobileNetV2()
299 std::generate(w89.begin(), w89.end() - size_t(sparsity * w89.size()), std::ref(f32rng)); in FP32SparseMobileNetV2()
305 std::generate(w93.begin(), w93.end() - size_t(sparsity * w93.size()), std::ref(f32rng)); in FP32SparseMobileNetV2()
[all …]
|
| D | fp32-sparse-mobilenet-v3-small.cc | 19 ExecutionPlan FP32SparseMobileNetV3Small(float sparsity, pthreadpool_t threadpool) { in FP32SparseMobileNetV3Small() argument
337 std::generate(w104.begin(), w104.end() - size_t(sparsity * w104.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Small()
341 std::generate(w106.begin(), w106.end() - size_t(sparsity * w106.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Small()
345 std::generate(w108.begin(), w108.end() - size_t(sparsity * w108.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Small()
349 std::generate(w110.begin(), w110.end() - size_t(sparsity * w110.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Small()
355 std::generate(w114.begin(), w114.end() - size_t(sparsity * w114.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Small()
359 std::generate(w116.begin(), w116.end() - size_t(sparsity * w116.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Small()
365 std::generate(w120.begin(), w120.end() - size_t(sparsity * w120.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Small()
369 std::generate(w122.begin(), w122.end() - size_t(sparsity * w122.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Small()
375 std::generate(w126.begin(), w126.end() - size_t(sparsity * w126.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Small()
[all …]
|
| D | fp32-sparse-mobilenet-v3-large.cc | 19 ExecutionPlan FP32SparseMobileNetV3Large(float sparsity, pthreadpool_t threadpool) { in FP32SparseMobileNetV3Large() argument
385 std::generate(w118.begin(), w118.end() - size_t(sparsity * w118.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Large()
389 std::generate(w120.begin(), w120.end() - size_t(sparsity * w120.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Large()
395 std::generate(w124.begin(), w124.end() - size_t(sparsity * w124.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Large()
399 std::generate(w126.begin(), w126.end() - size_t(sparsity * w126.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Large()
405 std::generate(w130.begin(), w130.end() - size_t(sparsity * w130.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Large()
409 std::generate(w132.begin(), w132.end() - size_t(sparsity * w132.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Large()
415 std::generate(w136.begin(), w136.end() - size_t(sparsity * w136.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Large()
419 std::generate(w138.begin(), w138.end() - size_t(sparsity * w138.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Large()
423 std::generate(w140.begin(), w140.end() - size_t(sparsity * w140.size()), std::ref(f32rng)); in FP32SparseMobileNetV3Large()
[all …]
|
| D | fp32-sparse-mobilenet-v1.cc | 19 ExecutionPlan FP32SparseMobileNetV1(float sparsity, pthreadpool_t threadpool) { in FP32SparseMobileNetV1() argument
145 std::generate(w34.begin(), w34.end() - size_t(sparsity * w34.size()), std::ref(f32rng)); in FP32SparseMobileNetV1()
151 std::generate(w38.begin(), w38.end() - size_t(sparsity * w38.size()), std::ref(f32rng)); in FP32SparseMobileNetV1()
157 std::generate(w42.begin(), w42.end() - size_t(sparsity * w42.size()), std::ref(f32rng)); in FP32SparseMobileNetV1()
163 std::generate(w46.begin(), w46.end() - size_t(sparsity * w46.size()), std::ref(f32rng)); in FP32SparseMobileNetV1()
169 std::generate(w50.begin(), w50.end() - size_t(sparsity * w50.size()), std::ref(f32rng)); in FP32SparseMobileNetV1()
175 std::generate(w54.begin(), w54.end() - size_t(sparsity * w54.size()), std::ref(f32rng)); in FP32SparseMobileNetV1()
181 std::generate(w58.begin(), w58.end() - size_t(sparsity * w58.size()), std::ref(f32rng)); in FP32SparseMobileNetV1()
187 std::generate(w62.begin(), w62.end() - size_t(sparsity * w62.size()), std::ref(f32rng)); in FP32SparseMobileNetV1()
193 std::generate(w66.begin(), w66.end() - size_t(sparsity * w66.size()), std::ref(f32rng)); in FP32SparseMobileNetV1()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | fully_connected.cc | 44 bool SupportedSparsityFormat(const TfLiteSparsity& sparsity) { in SupportedSparsityFormat() argument
45 if (sparsity.dim_metadata[0].format == kTfLiteDimDense && in SupportedSparsityFormat()
46 sparsity.dim_metadata[1].format == kTfLiteDimSparseCSR) { in SupportedSparsityFormat()
56 TfLiteStatus CreateLedgerTensor(const TfLiteSparsity* sparsity, in CreateLedgerTensor() argument
58 TF_LITE_ENSURE(context, sparsity != nullptr); in CreateLedgerTensor()
62 ledger_size->data[0] = sparsity->dim_metadata[1].array_indices->size + in CreateLedgerTensor()
63 sparsity->dim_metadata[1].array_segments->size - 1; in CreateLedgerTensor()
67 TfLiteStatus PopulateLedgerData(const TfLiteSparsity* sparsity, in PopulateLedgerData() argument
69 TF_LITE_ENSURE(context, sparsity != nullptr); in PopulateLedgerData()
70 const auto* array_segments = sparsity->dim_metadata[1].array_segments; in PopulateLedgerData()
[all …]
|
| D | densify.cc | 61 TF_LITE_ENSURE(context, op_context.input->sparsity != nullptr); in Prepare()
79 reference_ops::Densify(op_context.input->sparsity, in Eval()
87 op_context.input->sparsity, GetTensorShape(op_context.input), in Eval()
93 reference_ops::Densify(op_context.input->sparsity, in Eval()
|
| D | lstm.cc | 95 TfLiteStatus make_ledger(const TfLiteSparsity* sparsity, TfLiteContext* context, in make_ledger() argument
99 if (sparsity == nullptr) { in make_ledger()
103 ledger_size->data[0] = sparsity->dim_metadata[1].array_indices->size + in make_ledger()
104 sparsity->dim_metadata[1].array_segments->size - 1; in make_ledger()
108 TfLiteStatus copy_ledger(const TfLiteSparsity* sparsity, TfLiteTensor* ledger) { in copy_ledger() argument
109 if (sparsity == nullptr) { in copy_ledger()
113 const auto* array_segments = sparsity->dim_metadata[1].array_segments; in copy_ledger()
114 const auto* array_indices = sparsity->dim_metadata[1].array_indices; in copy_ledger()
1373 const bool is_sparse_op = (input_to_output_weights->sparsity != nullptr); in Prepare()
1635 : input_to_input_weights->sparsity, in Prepare()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| 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()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/ |
| 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dense_to_sparse.cc | 106 float sparsity = 0; in CalculateBlockSparsity() local
119 tflite::optimize::sparsity::FormatConverter<float> format_converter( in CalculateBlockSparsity()
125 sparsity = in CalculateBlockSparsity()
129 tflite::optimize::sparsity::FormatConverter<Eigen::half> format_converter( in CalculateBlockSparsity()
136 sparsity = in CalculateBlockSparsity()
140 tflite::optimize::sparsity::FormatConverter<int8_t> format_converter( in CalculateBlockSparsity()
146 sparsity = in CalculateBlockSparsity()
151 return sparsity; in CalculateBlockSparsity()
244 tflite::optimize::sparsity::FormatConverter<T> format_converter( in BuildSparsityParameterAttribute()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | sparse_add_op_test.py | 217 def _s2d_add_vs_sparse_add(sparsity, n, m, num_iters=50): argument
222 sp_t, unused_nnz = _sparsify(sp_vals, thresh=sparsity, index_dtype=np.int32)
246 for sparsity in [0.99, 0.5, 0.01]:
249 s2d_dt, sa_dt = _s2d_add_vs_sparse_add(sparsity, n, m)
250 print("%.2f \t %d \t %d \t %.4f \t %.4f \t %.2f" % (sparsity, n, m,
|