Home
last modified time | relevance | path

Searched defs:traversal_order (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/utils/
H A Dsparsity_format_converter_test.cc28 const std::vector<int> traversal_order = {0, 1}; in TEST() local
57 const std::vector<int> traversal_order = {0, 1}; in TEST() local
88 const std::vector<int> traversal_order = {0, 1}; in TEST() local
119 const std::vector<int> traversal_order = {0, 1}; in TEST() local
152 const std::vector<int> traversal_order = {1, 0}; in TEST() local
181 const std::vector<int> traversal_order = {1, 0}; in TEST() local
212 const std::vector<int> traversal_order = {1, 0}; in TEST() local
243 const std::vector<int> traversal_order = {1, 0}; in TEST() local
276 const std::vector<int> traversal_order = {0, 1, 2}; variable
312 const std::vector<int> traversal_order = {0, 1, 2}; variable
[all …]
H A Dsparsity_format_converter.cc56 const std::vector<int>& shape, const std::vector<int>& traversal_order, in FormatConverter()
219 const std::vector<int>& shape, const std::vector<int>& traversal_order, in FormatConverter()
232 auto traversal_order = TfLiteIntArrayToVector(sparsity.traversal_order); in FormatConverter() local
255 std::vector<int> shape, std::vector<int> traversal_order, in InitSparseToDenseConverter()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
H A Ddense_to_sparse.cc59 std::vector<int>* traversal_order, in PopulateEncodingParams()
119 std::vector<int> traversal_order = {}; in CalculateBlockSparsity() local
245 std::vector<int> traversal_order = {}; in BuildSparsityParameterAttribute() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dbinary_elementwise_tester.cc359 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
396 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
H A Dprelu_tester.cc204 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
H A Dconv_2d_tester.cc311 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
H A Ddepthwise_conv_2d_tester.cc320 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
H A Dtranspose_conv_tester.cc308 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel() local
/aosp_15_r20/external/ruy/ruy/
H A Dblock_map.h87 BlockMapTraversalOrder traversal_order; member
H A Dblock_map_test.cc150 BlockMapTraversalOrder traversal_order) { in GetBlockByIndexSquareTest()
/aosp_15_r20/external/tensorflow/tensorflow/lite/testdata/
H A Dsparse_tensor.json19 "traversal_order": [0, 1, 2, 3], array
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/
H A Dverifier.cc313 std::vector<int> traversal_order(total_dims); in VerifyAndCountSparseElements() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dtest_util.h145 std::vector<int> traversal_order; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dmodel_test.cc606 TfLiteIntArray* traversal_order = TfLiteIntArrayCreate(4); in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dcommon.h374 TfLiteIntArray* traversal_order; member
/aosp_15_r20/external/mesa3d/include/tensorflow/lite/core/c/
H A Dcommon.h387 TfLiteIntArray* traversal_order; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/
H A Dflatbuffer_export.cc2220 std::vector<int> traversal_order(dim_size); in BuildSparsityParameters() local