Home
last modified time | relevance | path

Searched refs:FlexModelTest (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/flex/
Dtest_util.cc26 bool FlexModelTest::Invoke() { return interpreter_->Invoke() == kTfLiteOk; } in Invoke()
28 void FlexModelTest::SetStringValues(int tensor_index, in SetStringValues()
38 std::vector<string> FlexModelTest::GetStringValues(int tensor_index) const { in GetStringValues()
51 void FlexModelTest::SetShape(int tensor_index, const std::vector<int>& values) { in SetShape()
56 std::vector<int> FlexModelTest::GetShape(int tensor_index) { in GetShape()
66 TfLiteType FlexModelTest::GetType(int tensor_index) { in GetType()
70 bool FlexModelTest::IsDynamicTensor(int tensor_index) { in IsDynamicTensor()
74 void FlexModelTest::AddTensors(int num_tensors, const std::vector<int>& inputs, in AddTensors()
95 void FlexModelTest::SetConstTensor(int tensor_index, in SetConstTensor()
107 void FlexModelTest::AddTfLiteMulOp(const std::vector<int>& inputs, in AddTfLiteMulOp()
[all …]
Dtest_util.h42 class FlexModelTest : public ::testing::Test {
44 FlexModelTest() {} in FlexModelTest() function
45 ~FlexModelTest() {} in ~FlexModelTest()
Ddelegate_test.cc31 class DelegateTest : public testing::FlexModelTest {
Dkernel_test.cc42 class KernelTest : public testing::FlexModelTest {