Home
last modified time | relevance | path

Searched defs:sparsity (Results 1 – 25 of 40) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
H A Dfully_connected.h34 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight()
80 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight1x16Impl()
116 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight1x4Impl()
200 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight1x16()
224 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight1x4()
/aosp_15_r20/external/pytorch/benchmarks/sparse/dlmc/
H A Dutils.py72 def load_spmv_dataset(dataset_path, hidden_size, sparsity, device, n_limit=math.inf): argument
111 dataset_path, hidden_size, sparsity, spmm_type, device, n_limit=math.inf argument
160 sparsity, argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dfully_connected.cc44 bool SupportedSparsityFormat(const TfLiteSparsity& sparsity) { in SupportedSparsityFormat()
56 TfLiteStatus CreateLedgerTensor(const TfLiteSparsity* sparsity, in CreateLedgerTensor()
68 TfLiteStatus PopulateLedgerData(const TfLiteSparsity* sparsity, in PopulateLedgerData()
919 const TfLiteSparsity* sparsity) { in VerifySparsity()
1005 const TfLiteSparsity& sparsity = *filter->sparsity; in EvalQuantized() local
1170 const auto& sparsity = *filter->sparsity; in EvalFloat() local
1191 const auto& sparsity = *filter->sparsity; in EvalFloat() local
/aosp_15_r20/external/XNNPACK/test/
H A Dspmm-microkernel-tester.h93 inline SpMMMicrokernelTester& sparsity(float sparsity) { in sparsity() argument
98 inline float sparsity() const { in sparsity() function
H A Dconvolution-operator-tester.h453 inline ConvolutionOperatorTester& sparsity(float sparsity) { in sparsity() argument
458 inline float sparsity() const { in sparsity() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
H A Dgemm-block-sparse-microkernel-tester.h36 float sparsity, in fillBlockSparseWeights()
141 inline GemmBlockSparseMicrokernelTester& sparsity(float s) { in sparsity() function
158 inline float sparsity() const { in sparsity() function
H A Dfully-connected-sparse-operator-tester.h34 float sparsity, in fillBlockSparseWeights()
175 inline FullyConnectedSparseOperatorTester& sparsity(float s) { in sparsity() function
188 inline float sparsity() const { in sparsity() function
/aosp_15_r20/external/pytorch/torch/sparse/
H A D_triton_ops_meta.py432 def create_blocked_tensor(B, M, N, blocksize, sparsity, dtype, device): argument
460 m, k, n, bm, bk, dtype=torch.float16, device="cuda", sparsity=0.5, force=False argument
730 sparsity=0.5, argument
/aosp_15_r20/external/pytorch/benchmarks/sparse/
H A Dtriton_ops.py5 def create_blocked_tensor(B, M, N, blocksize, sparsity, dtype, device): argument
308 sparsity = 1 - bk * bm * sparsity / (m * k) variable
318 sparsity = 1 - bk * bm * x._nnz() / (m * k) variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/utils/
H A Dsparsity_format_converter.cc24 namespace sparsity { namespace
231 const TfLiteSparsity& sparsity) { in FormatConverter()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dverifier.cc229 const SparsityParameters& sparsity, const std::vector<int>& dim_sizes) { in VerifyAndCountElements()
285 const auto* sparsity = tensor.sparsity(); in VerifyAndCountSparseElements() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter_builder.cc466 auto* sparsity = in ParseSparsity() local
642 TfLiteSparsity* sparsity = nullptr; in ParseTensors() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/
H A Dq8gemm_sparse.cc43 float sparsity, in fillBlockSparseWeights()
340 inline float sparsity() const { in sparsity() function in Q8GEMMSparse
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Ddensify.h29 inline void Densify(const TfLiteSparsity* sparsity, in Densify()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/
H A Dfully_connected.h26 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight()
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/
H A Dsparse_add_op_test.py249 def _s2d_add_vs_sparse_add(sparsity, n, m, num_iters=50): argument
/aosp_15_r20/external/pytorch/docs/source/
H A Dsparse.rst313 .. _sparse-coo-docs:
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dcommon.cc134 void TfLiteSparsityFree(TfLiteSparsity* sparsity) { in TfLiteSparsityFree()
/aosp_15_r20/external/XNNPACK/bench/
H A Df16-spmm.cc29 xnn_f16_spmm_minmax_ukernel_function spmm, uint32_t mr, uint32_t nr, float sparsity, in f16_spmm()
/aosp_15_r20/external/tensorflow/tensorflow/lite/testdata/
H A Dsparse_tensor.json18 "sparsity": { object
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Dgrafter.cc237 auto sparsity = CopyTable("tflite.SparsityParameters", i->sparsity()); in AdjustTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/model_modifier/
H A Dgrafter.cc236 auto sparsity = CopyTable("tflite.SparsityParameters", i->sparsity()); in AdjustTensors() local
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/
H A Dinit.cpp153 Tensor sparse_(Tensor tensor, double sparsity, double std) { in sparse_()
/aosp_15_r20/external/pytorch/torch/nn/
H A Dinit.py627 sparsity, argument
/aosp_15_r20/external/XNNPACK/models/
H A Dfp32-sparse-mobilenet-v1.cc21 ExecutionPlan FP32SparseMobileNetV1(float sparsity, pthreadpool_t threadpool) { in FP32SparseMobileNetV1()

12