Home
last modified time | relevance | path

Searched refs:subgraph_test_util (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Dcall_test.cc37 class CallTest : public subgraph_test_util::ControlFlowOpTest {
200 subgraph_test_util::FillIntTensor( in TEST_F()
202 subgraph_test_util::FillIntTensor( in TEST_F()
207 subgraph_test_util::CheckIntTensor(output, test_shapes[i], in TEST_F()
223 subgraph_test_util::CheckIntTensor(output, {0, 3}, {}); in TEST_F()
246 subgraph_test_util::FillIntTensor( in TEST_F()
248 subgraph_test_util::FillIntTensor( in TEST_F()
253 subgraph_test_util::CheckIntTensor(output, kCallOpInputShape, in TEST_F()
271 subgraph_test_util::FillIntTensor( in TEST_F()
273 subgraph_test_util::FillIntTensor( in TEST_F()
[all …]
H A DBUILD55 "//tensorflow/lite/kernels:subgraph_test_util",
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dwhile_test.cc27 using subgraph_test_util::CheckIntTensor;
28 using subgraph_test_util::CheckScalarStringTensor;
29 using subgraph_test_util::CheckStringTensor;
30 using subgraph_test_util::ControlFlowOpTest;
31 using subgraph_test_util::FillIntTensor;
32 using subgraph_test_util::FillScalarStringTensor;
H A Dif_test.cc28 using subgraph_test_util::CheckIntTensor;
29 using subgraph_test_util::ControlFlowOpTest;
30 using subgraph_test_util::FillIntTensor;
H A Dsubgraph_test_util.h35 namespace subgraph_test_util {
H A DBUILD2463 ":subgraph_test_util",
2478 ":subgraph_test_util",
2491 ":subgraph_test_util",
2601 name = "subgraph_test_util",
2603 srcs = ["subgraph_test_util.cc"],
2604 hdrs = ["subgraph_test_util.h"],
2624 ":subgraph_test_util",
H A Dcall_once_test.cc28 using subgraph_test_util::ControlFlowOpTest;
H A Dsubgraph_test_util_test.cc30 namespace subgraph_test_util { namespace
H A DCMakeLists.txt101 subgraph_test_util.cc
H A Dsubgraph_test_util.cc72 namespace subgraph_test_util { namespace
/aosp_15_r20/external/tensorflow/tensorflow/lite/profiling/
H A Dprofile_summarizer_test.cc154 class ProfileSummarizerIfOpTest : public subgraph_test_util::ControlFlowOpTest {
167 subgraph_test_util::FillIntTensor( in SetUp()
169 subgraph_test_util::FillIntTensor( in SetUp()
183 subgraph_test_util::CheckIntTensor(output, {1, 2}, {6, 9}); in TEST_F()
210 subgraph_test_util::CheckIntTensor(output, {1, 2}, {5, 14}); in TEST_F()
H A DBUILD245 "//tensorflow/lite/kernels:subgraph_test_util",
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Ddelegate_data_test.cc78 subgraph_test_util::SubgraphBuilder builder; in TEST()
223 subgraph_test_util::SubgraphBuilder builder; in TEST()
H A DBUILD244 "//tensorflow/lite/kernels:subgraph_test_util",
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/serialization/
H A Dwriter_lib_test.cc41 using subgraph_test_util::CheckIntTensor;
42 using subgraph_test_util::FillIntTensor;
495 class WhileTest : public subgraph_test_util::ControlFlowOpTest {
H A DBUILD74 "//tensorflow/lite/kernels:subgraph_test_util",