- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for CreateClusterOutliningPass (0.27 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
pm.addNestedPass<FuncOp>( tensorflow::tf2xla::internal::CreateHoistBroadcastReadPass()); pm.addNestedPass<FuncOp>( tensorflow::tf2xla::internal::CreateXlaBroadcastPass()); pm.addPass(mlir::TFDevice::CreateClusterOutliningPass()); pm.addPass(mlir::TFTPU::CreateTPUResourceReadForWritePass()); pm.addPass(mlir::TFDevice::CreateMarkInputOutputAliasesPass()); pm.addPass(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 16:09:14 UTC 2024 - 11.2K bytes - Viewed (1) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_outlining.cc
module.walk([&](tf_device::LaunchOp launch) { OutlineLaunch(launch, &symbol_table, &builder); }); } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateClusterOutliningPass() { return std::make_unique<ClusterOutliningPass>(); } std::unique_ptr<OperationPass<ModuleOp>> CreateLaunchOutliningPass() { return std::make_unique<LaunchOutliningPass>(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
llvm::function_ref<bool(tf_device::ClusterOp, ElementsAttr)> filter = {}); // Creates a pass that outlines regions of tf_device.cluster operations. std::unique_ptr<OperationPass<ModuleOp>> CreateClusterOutliningPass(); // Creates a pass that outlines regions of tf_device.launch operations. std::unique_ptr<OperationPass<ModuleOp>> CreateLaunchOutliningPass(); // Creates a pass that converts tf_device::LaunchFuncOp into
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td
%identity = "tf.Identity"(%arg0) : (tensor<i32>) -> tensor<i32> return %identity : tensor<i32> } ``` }]; let constructor = "TFDevice::CreateClusterOutliningPass()"; } def ConvertTfControlFlowToScfPass : Pass<"convert-tf-control-flow-to-scf", "ModuleOp"> { let summary = "Convert TensorFlow control flow to SCF."; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)