- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for CreateResourceOpLiftingPass (0.36 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
pm.addPass(mlir::TF::CreateTFShapeInferencePass()); pm.addNestedPass<FuncOp>(mlir::createCanonicalizerPass()); pm.addPass(mlir::TFTPU::CreateTPUClusterCleanupAttributesPass()); pm.addPass(mlir::TFDevice::CreateResourceOpLiftingPass()); // Re-run the canonicalizer pass as some cleanup during resource op lifting // pass opens up some opportunities for canonicalization of cluster ops.
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/tf_device_passes.td
%0 = "tf.Const"() ... return %0 : tensor<f32> } func @while_cond(%arg0: tensor<f32>) { return %arg0 } ``` }]; let constructor = "TFDevice::CreateResourceOpLiftingPass()"; } def ResourceOpLiftingForMainFunctionPass : Pass<"tf-resource-op-lifting-for-main-function", "ModuleOp"> { let summary = "Lifting resource operations out of control flow statements "
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
// variable store operations are all after device computation. After this pass, // device computation no longer interacts with external resource variables. std::unique_ptr<OperationPass<ModuleOp>> CreateResourceOpLiftingPass(); // Creates a pass that lifts operations from the main function. std::unique_ptr<OperationPass<ModuleOp>> CreateResourceOpLiftingForMainFunctionPass();
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/resource_op_lifting.cc
return signalPassFailure(); } } } // namespace namespace TFDevice { std::unique_ptr<OperationPass<ModuleOp>> CreateResourceOpLiftingPass() { return std::make_unique<ResourceOpLiftingPass>(); } std::unique_ptr<OperationPass<ModuleOp>> CreateResourceOpLiftingForMainFunctionPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0)