- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for CreateTPURewritePass (0.2 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/runtime_passes.h
namespace mlir { namespace TFTPU { // Creates a pass that rewrites `tf_device.launch_func` on TPUs into TPU runtime // ops. std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateTPURewritePass( llvm::StringRef module_name = llvm::StringRef()); // Creates a pass that adds ops which perform formatting on variables at // run-time according to compilation result. std::unique_ptr<OperationPass<ModuleOp>>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 10 18:58:57 UTC 2024 - 2.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops.cc
} } // namespace void AddTPULowerClusterToRuntimeOpsPassPipeline(OpPassManager& pm, llvm::StringRef module_name) { pm.addPass(mlir::TFTPU::CreateTPURewritePass(module_name)); pm.addPass(mlir::createSymbolDCEPass()); pm.addNestedPass<FuncOp>( mlir::TFDevice::CreateReplicateInvariantOpHoistingPass());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 17 18:52:57 UTC 2024 - 9.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/runtime_passes.td
]; let dependentDialects = [ "mlir::mhlo::MhloDialect", "mlir::tf_device::TensorFlowDeviceDialect" ]; let constructor = "mlir::TFTPU::CreateTPURewritePass()"; } def TPUVariableRuntimeReformattingPass : Pass<"tf-tpu-variable-runtime-reformatting", "ModuleOp"> { let summary = "Adds device variable formatting op to allow compilation-guided "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 10 18:58:57 UTC 2024 - 10.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
} op.erase(); } } // TODO(b/139377366): Remove functions that are no longer needed. } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateTPURewritePass( llvm::StringRef module_name) { return std::make_unique<TPURewritePass>(module_name); } } // namespace TFTPU
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)