Searched refs:TfLiteDimensionType (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/python/ |
| D | format_converter_extension.cc | 40 py::enum_<TfLiteDimensionType>(m, "TfLiteDimensionType") in PYBIND11_MODULE()
41 .value("TF_LITE_DIM_DENSE", TfLiteDimensionType::kTfLiteDimDense) in PYBIND11_MODULE()
42 .value("TF_LITE_DIM_SPARSE_CSR", TfLiteDimensionType::kTfLiteDimSparseCSR) in PYBIND11_MODULE()
48 /*format=*/const std::vector<TfLiteDimensionType>&, in PYBIND11_MODULE()
|
| D | format_converter_extension_test.py | 41 format_converter.TfLiteDimensionType.TF_LITE_DIM_DENSE,
42 format_converter.TfLiteDimensionType.TF_LITE_DIM_SPARSE_CSR
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter_test.cc | 29 const std::vector<TfLiteDimensionType> format = {kTfLiteDimDense, in TEST()
58 const std::vector<TfLiteDimensionType> format = {kTfLiteDimSparseCSR, in TEST()
89 const std::vector<TfLiteDimensionType> format = {kTfLiteDimDense, in TEST()
120 const std::vector<TfLiteDimensionType> format = {kTfLiteDimSparseCSR, in TEST()
153 const std::vector<TfLiteDimensionType> format = {kTfLiteDimDense, in TEST()
182 const std::vector<TfLiteDimensionType> format = {kTfLiteDimDense, in TEST()
213 const std::vector<TfLiteDimensionType> format = {kTfLiteDimSparseCSR, in TEST()
244 const std::vector<TfLiteDimensionType> format = {kTfLiteDimSparseCSR, in TEST()
277 const std::vector<TfLiteDimensionType> format = {
313 const std::vector<TfLiteDimensionType> format = {
[all …]
|
| D | format_converter.h | 45 const std::vector<TfLiteDimensionType>& format,
92 std::vector<TfLiteDimensionType> format_;
|
| D | format_converter.cc | 55 const std::vector<TfLiteDimensionType>& format, in FormatConverter()
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 347 typedef enum TfLiteDimensionType { enum
350 } TfLiteDimensionType; typedef
354 TfLiteDimensionType format;
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dense_to_sparse.cc | 52 std::vector<TfLiteDimensionType>* format, in PopulateEncodingParams()
112 std::vector<TfLiteDimensionType> format = {}; in CalculateBlockSparsity()
238 std::vector<TfLiteDimensionType> format = {}; in BuildSparsityParameterAttribute()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.h | 115 std::vector<TfLiteDimensionType> format = {},
146 std::vector<TfLiteDimensionType> format;
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_builder.cc | 473 static_cast<TfLiteDimensionType>(src_metadata->format()); in ParseSparsity()
|