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.cc715 EXPECT_FALSE(ShouldExportAsFlexOp(false, "Conv2D")); in TEST_F()
716 EXPECT_TRUE(ShouldExportAsFlexOp(true, "Conv2D")); in TEST_F()
717 EXPECT_TRUE(ShouldExportAsFlexOp(true, "EluGrad")); in TEST_F()
718 EXPECT_TRUE(ShouldExportAsFlexOp(true, "RFFT")); in TEST_F()
719 EXPECT_FALSE(ShouldExportAsFlexOp(true, "MyAwesomeCustomOp")); in TEST_F()
722 EXPECT_FALSE(ShouldExportAsFlexOp(true, "RandomShuffle")); in TEST_F()
Dexport.cc143 if (ShouldExportAsFlexOp(enable_select_tf_ops, in OperatorKey()
Doperator.cc1567 if (ShouldExportAsFlexOp(enable_select_tf_ops_, node_def.op())) { in WriteOptions()
2036 bool ShouldExportAsFlexOp(bool enable_select_tf_ops, in ShouldExportAsFlexOp() function