- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for CreateLaunchToDeviceAttributePass (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/launch_to_device_attribute.cc
mlir::TF::SplitIsland(curr_island, control_type); }); } } } // anonymous namespace std::unique_ptr<OperationPass<func::FuncOp>> CreateLaunchToDeviceAttributePass( bool legacy_graph_export) { return std::make_unique<LaunchToDeviceAttributePass>(legacy_graph_export); } } // namespace TFDevice
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v2/tf_dialect_to_executor.cc
pm.addNestedPass<FuncOp>(mlir::TFDevice::CreateParallelExecuteToIslandsPass( /*legacy_graph_export=*/false)); pm.addNestedPass<FuncOp>(mlir::TFDevice::CreateLaunchToDeviceAttributePass( /*legacy_graph_export=*/false)); // Do a single pass to encode necessary control deps in the IR according to // the results of side effect analysis. pm.addPass(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 13 23:22:50 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/api/v1/tf_dialect_to_executor.cc
add_pass(mlir::TFDevice::CreateReplicaIDToDeviceOrdinalPass()); add_pass(mlir::TFDevice::CreateParallelExecuteToIslandsPass( /*legacy_graph_export=*/true)); add_pass(mlir::TFDevice::CreateLaunchToDeviceAttributePass( /*legacy_graph_export=*/true)); pm.addNestedPass<FuncOp>(mlir::TFTPU::CreateTPUDevicePropagationPass()); pm.addNestedPass<FuncOp>(mlir::TFTPU::CreateTPUColocateSplitsPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 16 03:41:02 UTC 2024 - 7.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
// come before TPUClusterCleanupAttributes pass or else the device attribute // will be removed from launch causing an error. pm.addNestedPass<FuncOp>(mlir::TFDevice::CreateLaunchToDeviceAttributePass()); // Note that the region-based control-flow produced here still contains // function call ops which get inlined by the subsequent inliner pass.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 16:09:14 UTC 2024 - 11.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_device_passes.td
"islands created by this pass should be split after the replicated " "ops have been lifted."> ]; let constructor = "TFDevice::CreateLaunchToDeviceAttributePass()"; } def XlaInlineDeviceOpsPass : Pass<"tf-xla-inline-device-ops", "ModuleOp"> { let summary = "Inline all Cluster op based in the parent region";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 12.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
// attribute to each TensorFlow dialect op in the body based on the `device` // attribute on the `tf_device.launch`. std::unique_ptr<OperationPass<func::FuncOp>> CreateLaunchToDeviceAttributePass( bool legacy_graph_export = true); // Creates a pass to ensure that the `_xla_outside_compilation` and // tf_device.launch op no longer exist after Outside Compilation is complete.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 31.8K bytes - Viewed (0)