- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for legalizeTFPasses (0.3 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/xla_legalize_tf_test.cc
PassManager pm(&context); if (mlir::failed(mlir::applyPassManagerCLOptions(pm))) return false; passes(&pm); return pm.run(module.get()).succeeded(); } std::function<void(PassManager*)> legalizeTFPasses() { return [](PassManager* pm) { pm->addPass(mlir::mhlo::createLegalizeTFPass( /* legalize_chlo=*/true, llvm::StringRef("gpu/xpu"), /* prefer_tf2xla=*/false)); }; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:16:07 UTC 2024 - 4.2K bytes - Viewed (0)