Searched refs:TfLiteSparsity (Results 1 – 16 of 16) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/ |
D | fully_connected.h | 31 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight() 77 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight1x4Impl() 117 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight1x4Task() 145 const TfLiteSparsity& sparsity; 165 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight1x4()
|
/external/tensorflow/tensorflow/lite/c/ |
D | common.h | 362 typedef struct TfLiteSparsity { struct 367 } TfLiteSparsity; argument 437 TfLiteSparsity* sparsity; 590 void TfLiteSparsityFree(TfLiteSparsity* sparsity);
|
D | common_test.cc | 123 t.sparsity = static_cast<TfLiteSparsity*>(malloc(sizeof(TfLiteSparsity))); in TEST()
|
D | common.c | 109 void TfLiteSparsityFree(TfLiteSparsity* sparsity) { in TfLiteSparsityFree()
|
/external/tensorflow/tensorflow/lite/ |
D | interpreter_builder.cc | 423 const SparsityParameters* src_sparsity, TfLiteSparsity** sparsity_ptr) { in ParseSparsity() 435 reinterpret_cast<TfLiteSparsity*>(malloc(sizeof(TfLiteSparsity))); in ParseSparsity() 436 memset(sparsity, 0, sizeof(TfLiteSparsity)); in ParseSparsity() 614 TfLiteSparsity* sparsity = nullptr; in ParseTensors()
|
D | interpreter_builder.h | 99 TfLiteSparsity** sparsity);
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | densify.h | 29 inline void Densify(const TfLiteSparsity* 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()
|
/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/python/ |
D | format_converter_extension.cc | 52 /*sparsity=*/const TfLiteSparsity&>()) in PYBIND11_MODULE()
|
/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
D | format_converter.h | 54 const TfLiteSparsity& sparsity);
|
D | format_converter.cc | 217 const TfLiteSparsity& sparsity) in FormatConverter()
|
/external/tensorflow/tensorflow/lite/core/ |
D | subgraph.h | 106 TfLiteSparsity* sparsity = nullptr) { 115 TfLiteSparsity* sparsity = nullptr);
|
D | subgraph.cc | 60 void operator()(TfLiteSparsity* s) { in operator ()() 66 std::unique_ptr<TfLiteSparsity, TfLiteSparsityDeleter>; 1196 size_t bytes, const Allocation* allocation, TfLiteSparsity* sparsity) { in SetTensorParametersReadOnly()
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | fully_connected.cc | 44 bool SupportedSparsityFormat(const TfLiteSparsity& sparsity) { in SupportedSparsityFormat() 56 TfLiteStatus CreateLedgerTensor(const TfLiteSparsity* sparsity, in CreateLedgerTensor() 67 TfLiteStatus PopulateLedgerData(const TfLiteSparsity* sparsity, in PopulateLedgerData()
|
D | lstm.cc | 95 TfLiteStatus make_ledger(const TfLiteSparsity* sparsity, TfLiteContext* context, in make_ledger() 108 TfLiteStatus copy_ledger(const TfLiteSparsity* sparsity, TfLiteTensor* ledger) { in copy_ledger()
|
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
D | interpreter_wrapper.cc | 108 PyObject* PyDictFromSparsityParam(const TfLiteSparsity& param) { in PyDictFromSparsityParam()
|