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 | 73 input.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST()
87 input.format = {kTfLiteDimDense, kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST()
101 input.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST()
|
| D | fully_connected_test.cc | 1254 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1279 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1306 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1336 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1369 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1428 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
1484 weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_P()
|
| D | test_util.h | 237 t.format[t.traversal_order[i]] == kTfLiteDimSparseCSR) { in AddConstSparseInput()
309 t.format[t.traversal_order[i]] == kTfLiteDimSparseCSR) {
|
| D | lstm_test.cc | 2648 input_weight.format = {kTfLiteDimDense, kTfLiteDimSparseCSR}; in TEST_F()
2655 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 | 128 if (metadata.format == kTfLiteDimSparseCSR) { in TfLiteSparsityFree()
|
| D | common.h | 349 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 | 446 ASSERT_EQ(t1->sparsity->dim_metadata[1].format, kTfLiteDimSparseCSR); in TEST()
|