Home
last modified time | relevance | path

Searched refs:IsFlexOp (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/
Dutil_test.cc44 TEST(UtilTest, IsFlexOp) { in TEST() argument
45 EXPECT_TRUE(IsFlexOp("Flex")); in TEST()
46 EXPECT_TRUE(IsFlexOp("FlexOp")); in TEST()
47 EXPECT_FALSE(IsFlexOp("flex")); in TEST()
48 EXPECT_FALSE(IsFlexOp("Fle")); in TEST()
49 EXPECT_FALSE(IsFlexOp("OpFlex")); in TEST()
50 EXPECT_FALSE(IsFlexOp(nullptr)); in TEST()
51 EXPECT_FALSE(IsFlexOp("")); in TEST()
Dutil.h37 bool IsFlexOp(const char* custom_name);
Dutil.cc21 bool IsFlexOp(const char* custom_name) { in IsFlexOp() function
Dmodel.cc450 IsFlexOp(registration->custom_name)) { in ApplyDelegates()
/external/tensorflow/tensorflow/lite/delegates/flex/
Ddelegate.cc60 if (IsFlexOp(registration->custom_name)) { in Prepare()
/external/tensorflow/tensorflow/lite/kernels/
Dregister_ref.cc154 if (IsFlexOp(op)) { in FindOp()
Dregister.cc160 if (IsFlexOp(op)) { in FindOp()