|
|
|
@ -12,13 +12,14 @@
|
|
|
|
|
// See the License for the specific language governing permissions and
|
|
|
|
|
// limitations under the License.
|
|
|
|
|
|
|
|
|
|
#include "paddle/fluid/inference/analysis/analyzer.h"
|
|
|
|
|
#include <gflags/gflags.h>
|
|
|
|
|
#include <glog/logging.h> // use glog instead of PADDLE_ENFORCE to avoid importing other paddle header files.
|
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
|
#include "paddle/fluid/framework/ir/pass.h"
|
|
|
|
|
#include "paddle/fluid/inference/analysis/analyzer.h"
|
|
|
|
|
#include "paddle/fluid/inference/analysis/ut_helper.h"
|
|
|
|
|
#include "paddle/fluid/inference/api/paddle_inference_api.h"
|
|
|
|
|
#include "paddle/fluid/inference/api/paddle_inference_pass.h"
|
|
|
|
|
#include "paddle/fluid/inference/api/timer.h"
|
|
|
|
|
|
|
|
|
|
DEFINE_string(infer_model, "", "Directory of the inference model.");
|
|
|
|
@ -100,10 +101,3 @@ void Main(int batch_size) {
|
|
|
|
|
TEST(text_classification, basic) { Main(FLAGS_batch_size); }
|
|
|
|
|
|
|
|
|
|
} // namespace paddle
|
|
|
|
|
|
|
|
|
|
USE_PASS(fc_fuse_pass);
|
|
|
|
|
USE_PASS(seq_concat_fc_fuse_pass);
|
|
|
|
|
USE_PASS(fc_lstm_fuse_pass);
|
|
|
|
|
USE_PASS(graph_viz_pass);
|
|
|
|
|
USE_PASS(infer_clean_graph_pass);
|
|
|
|
|
USE_PASS(attention_lstm_fuse_pass);
|