Searched refs:kTfLiteDimDense (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter_test.cc | 29 const std::vector<TfLiteDimensionType> format = {kTfLiteDimDense, in TEST()
30 kTfLiteDimDense}; in TEST()
59 kTfLiteDimDense}; in TEST()
89 const std::vector<TfLiteDimensionType> format = {kTfLiteDimDense, in TEST()
153 const std::vector<TfLiteDimensionType> format = {kTfLiteDimDense, in TEST()
154 kTfLiteDimDense}; in TEST()
182 const std::vector<TfLiteDimensionType> format = {kTfLiteDimDense, in TEST()
214 kTfLiteDimDense}; in TEST()
278 kTfLiteDimSparseCSR, kTfLiteDimDense, kTfLiteDimSparseCSR};
314 kTfLiteDimDense, kTfLiteDimDense, kTfLiteDimSparseCSR};
[all …]
|
| D | format_converter.cc | 78 format_[i + shape.size()] = kTfLiteDimDense; in FormatConverter()
141 if (format_[i] == kTfLiteDimDense) { in DenseToSparse()
171 } else if (format_[num_expanded_dims - 1] == kTfLiteDimDense) { in DenseToSparse()
231 if (format_[i] == kTfLiteDimDense) { in FormatConverter()
288 if (format_[level] == kTfLiteDimDense) { in Populate()
|
| /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 | 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 | 45 if (sparsity.dim_metadata[0].format == kTfLiteDimDense && in SupportedSparsityFormat()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/python/ |
| D | format_converter_extension.cc | 41 .value("TF_LITE_DIM_DENSE", TfLiteDimensionType::kTfLiteDimDense) in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dense_to_sparse.cc | 61 (*format)[i] = kTfLiteDimDense; in PopulateEncodingParams()
70 format->push_back(kTfLiteDimDense); in PopulateEncodingParams()
252 if (format[i] == kTfLiteDimDense) { in BuildSparsityParameterAttribute()
|
| /external/tensorflow/tensorflow/lite/ |
| D | model_test.cc | 441 ASSERT_EQ(t1->sparsity->dim_metadata[0].format, kTfLiteDimDense); in TEST()
464 ASSERT_EQ(t1->sparsity->dim_metadata[2].format, kTfLiteDimDense); in TEST()
469 ASSERT_EQ(t1->sparsity->dim_metadata[3].format, kTfLiteDimDense); in TEST()
|
| D | interpreter_builder.cc | 475 if (tgt_metadata->format == kTfLiteDimDense) { in ParseSparsity()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common_test.cc | 131 t.sparsity->dim_metadata[0].format = kTfLiteDimDense; in TEST()
|
| D | common.h | 348 kTfLiteDimDense = 0, enumerator
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.cc | 119 if (param.dim_metadata[i].format == kTfLiteDimDense) { in PyDictFromSparsityParam()
|