Home
last modified time | relevance | path

Searched refs:TfLiteSparsity (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h31 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/
Dcommon.h362 typedef struct TfLiteSparsity { struct
367 } TfLiteSparsity; argument
437 TfLiteSparsity* sparsity;
590 void TfLiteSparsityFree(TfLiteSparsity* sparsity);
Dcommon_test.cc123 t.sparsity = static_cast<TfLiteSparsity*>(malloc(sizeof(TfLiteSparsity))); in TEST()
Dcommon.c109 void TfLiteSparsityFree(TfLiteSparsity* sparsity) { in TfLiteSparsityFree()
/external/tensorflow/tensorflow/lite/
Dinterpreter_builder.cc423 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()
Dinterpreter_builder.h99 TfLiteSparsity** sparsity);
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddensify.h29 inline void Densify(const TfLiteSparsity* sparsity, in Densify()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/sparse_ops/
Dfully_connected.h26 const TfLiteSparsity& sparsity, const FullyConnectedParams& params, in FullyConnectedSparseWeight()
/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/python/
Dformat_converter_extension.cc52 /*sparsity=*/const TfLiteSparsity&>()) in PYBIND11_MODULE()
/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/
Dformat_converter.h54 const TfLiteSparsity& sparsity);
Dformat_converter.cc217 const TfLiteSparsity& sparsity) in FormatConverter()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h106 TfLiteSparsity* sparsity = nullptr) {
115 TfLiteSparsity* sparsity = nullptr);
Dsubgraph.cc60 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/
Dfully_connected.cc44 bool SupportedSparsityFormat(const TfLiteSparsity& sparsity) { in SupportedSparsityFormat()
56 TfLiteStatus CreateLedgerTensor(const TfLiteSparsity* sparsity, in CreateLedgerTensor()
67 TfLiteStatus PopulateLedgerData(const TfLiteSparsity* sparsity, in PopulateLedgerData()
Dlstm.cc95 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/
Dinterpreter_wrapper.cc108 PyObject* PyDictFromSparsityParam(const TfLiteSparsity& param) { in PyDictFromSparsityParam()