Searched refs:kTfLiteDimSparseCSR (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter_test.cc | 58 const std::vector<TfLiteDimensionType> format = {kTfLiteDimSparseCSR, in TEST()
90 kTfLiteDimSparseCSR}; in TEST()
120 const std::vector<TfLiteDimensionType> format = {kTfLiteDimSparseCSR, in TEST()
121 kTfLiteDimSparseCSR}; in TEST()
183 kTfLiteDimSparseCSR}; in TEST()
213 const std::vector<TfLiteDimensionType> format = {kTfLiteDimSparseCSR, in TEST()
244 const std::vector<TfLiteDimensionType> format = {kTfLiteDimSparseCSR, in TEST()
245 kTfLiteDimSparseCSR}; in TEST()
278 kTfLiteDimSparseCSR, kTfLiteDimDense, kTfLiteDimSparseCSR};
314 kTfLiteDimDense, kTfLiteDimDense, kTfLiteDimSparseCSR};
[all …]
|
| D | format_converter.cc | 125 if (format_[i] == kTfLiteDimSparseCSR) { in DenseToSparse()
182 } else if (format_[dst_dim_idx] == kTfLiteDimSparseCSR) { in DenseToSparse()
201 if (format_[dst_dim_idx] == kTfLiteDimSparseCSR) { in DenseToSparse()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | densify_test.cc | 75 input.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST()
89 input.format = {kTfLiteDimDense, kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST()
103 input.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST()
|
| D | fully_connected_test.cc | 1290 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1317 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1342 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1369 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1399 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1428 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1461 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1520 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1577 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
|
| D | test_util.h | 238 t.format[t.traversal_order[i]] == kTfLiteDimSparseCSR) { in AddConstSparseInput()
310 t.format[t.traversal_order[i]] == kTfLiteDimSparseCSR) {
|
| D | lstm_test.cc | 2646 input_weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_F()
2653 recurrent_weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_F()
|
| D | fully_connected.cc | 46 sparsity.dim_metadata[1].format == kTfLiteDimSparseCSR) { in SupportedSparsityFormat()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/python/ |
| D | format_converter_extension.cc | 42 .value("TF_LITE_DIM_SPARSE_CSR", TfLiteDimensionType::kTfLiteDimSparseCSR) in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common_test.cc | 134 t.sparsity->dim_metadata[1].format = kTfLiteDimSparseCSR; in TEST()
|
| D | common.c | 130 if (metadata.format == kTfLiteDimSparseCSR) { in TfLiteSparsityFree()
|
| D | common.h | 350 kTfLiteDimSparseCSR, enumerator
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dense_to_sparse.cc | 63 (*format)[dims_count - 1] = kTfLiteDimSparseCSR; in PopulateEncodingParams()
|
| /external/tensorflow/tensorflow/lite/ |
| D | model_test.cc | 607 ASSERT_EQ(t1->sparsity->dim_metadata[1].format, kTfLiteDimSparseCSR); in TEST()
|