Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.h125 bool ShouldExportAsFlexOp(bool enable_select_tf_ops,
Doperator_test.cc736 EXPECT_FALSE(ShouldExportAsFlexOp(false, "Conv2D")); in TEST_F()
737 EXPECT_TRUE(ShouldExportAsFlexOp(true, "Conv2D")); in TEST_F()
738 EXPECT_TRUE(ShouldExportAsFlexOp(true, "EluGrad")); in TEST_F()
739 EXPECT_TRUE(ShouldExportAsFlexOp(true, "RFFT")); in TEST_F()
740 EXPECT_FALSE(ShouldExportAsFlexOp(true, "MyAwesomeCustomOp")); in TEST_F()
743 EXPECT_FALSE(ShouldExportAsFlexOp(true, "RandomShuffle")); in TEST_F()
Dexport.cc144 if (ShouldExportAsFlexOp(enable_select_tf_ops, in OperatorKey()
Doperator.cc1647 if (ShouldExportAsFlexOp(enable_select_tf_ops_, node_def.op())) { in WriteOptions()
2115 bool ShouldExportAsFlexOp(bool enable_select_tf_ops, in ShouldExportAsFlexOp() function