/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
H A D | fully_connected.h | 34 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 D | utils.py | 72 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 D | fully_connected.cc | 44 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 D | spmm-microkernel-tester.h | 93 inline SpMMMicrokernelTester& sparsity(float sparsity) { in sparsity() argument 98 inline float sparsity() const { in sparsity() function
|
H A D | convolution-operator-tester.h | 453 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 D | gemm-block-sparse-microkernel-tester.h | 36 float sparsity, in fillBlockSparseWeights() 141 inline GemmBlockSparseMicrokernelTester& sparsity(float s) { in sparsity() function 158 inline float sparsity() const { in sparsity() function
|
H A D | fully-connected-sparse-operator-tester.h | 34 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.py | 432 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 D | triton_ops.py | 5 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 D | sparsity_format_converter.cc | 24 namespace sparsity { namespace 231 const TfLiteSparsity& sparsity) { in FormatConverter()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/ |
H A D | verifier.cc | 229 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 D | interpreter_builder.cc | 466 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 D | q8gemm_sparse.cc | 43 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 D | densify.h | 29 inline void Densify(const TfLiteSparsity* sparsity, in Densify()
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/ |
H A D | fully_connected.h | 26 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight()
|
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/sparse_ops/ |
H A D | sparse_add_op_test.py | 249 def _s2d_add_vs_sparse_add(sparsity, n, m, num_iters=50): argument
|
/aosp_15_r20/external/pytorch/docs/source/ |
H A D | sparse.rst | 313 .. _sparse-coo-docs:
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
H A D | common.cc | 134 void TfLiteSparsityFree(TfLiteSparsity* sparsity) { in TfLiteSparsityFree()
|
/aosp_15_r20/external/XNNPACK/bench/ |
H A D | f16-spmm.cc | 29 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 D | sparse_tensor.json | 18 "sparsity": { object
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/ |
H A D | grafter.cc | 237 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 D | grafter.cc | 236 auto sparsity = CopyTable("tflite.SparsityParameters", i->sparsity()); in AdjustTensors() local
|
/aosp_15_r20/external/pytorch/torch/csrc/api/src/nn/ |
H A D | init.cpp | 153 Tensor sparse_(Tensor tensor, double sparsity, double std) { in sparse_()
|
/aosp_15_r20/external/pytorch/torch/nn/ |
H A D | init.py | 627 sparsity, argument
|
/aosp_15_r20/external/XNNPACK/models/ |
H A D | fp32-sparse-mobilenet-v1.cc | 21 ExecutionPlan FP32SparseMobileNetV1(float sparsity, pthreadpool_t threadpool) { in FP32SparseMobileNetV1()
|