Home
last modified time | relevance | path

Searched refs:GetDimMetadata (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/utils/
Dsparsity_format_converter_test.cc34 const auto& dim_metadata = converter.GetDimMetadata(); in TEST()
63 const auto& dim_metadata = converter.GetDimMetadata(); in TEST()
94 const auto& dim_metadata = converter.GetDimMetadata(); in TEST()
125 const auto& dim_metadata = converter.GetDimMetadata(); in TEST()
158 const auto& dim_metadata = converter.GetDimMetadata(); in TEST()
187 const auto& dim_metadata = converter.GetDimMetadata(); in TEST()
218 const auto& dim_metadata = converter.GetDimMetadata(); in TEST()
249 const auto& dim_metadata = converter.GetDimMetadata(); in TEST()
282 const auto& dim_metadata = converter.GetDimMetadata();
318 const auto& dim_metadata = converter.GetDimMetadata();
[all …]
Dsparsity_format_converter.h81 const std::vector<std::vector<int>>& GetDimMetadata() { in GetDimMetadata() function
/external/tensorflow/tensorflow/lite/tools/optimize/sparsity/
Dformat_converter_wrapper_pybind11.cc49 .def("GetDimMetadata", &FormatConverterFp32::GetDimMetadata) in PYBIND11_MODULE()
Dformat_converter_wrapper_pybind11_test.py47 dim_metadata = converter.GetDimMetadata()
/external/tensorflow/tensorflow/lite/kernels/
Dtest_util.h260 const auto& dim_metadata = converter.GetDimMetadata(); in AddConstSparseInput()
332 const auto dim_metadata = converter.GetDimMetadata();
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddense_to_sparse.cc255 const auto& metadata = format_converter.GetDimMetadata(); in BuildSparsityParameterAttribute()