- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for CreateResourceDeviceInferencePass (0.26 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/clustering_bridge_passes.cc
pm.addNestedPass<FuncOp>( mlir::TFDevice::CreateVerifyNoOutsideCompilationMarkersPass()); pm.addNestedPass<FuncOp>(mlir::TFDevice::CreateClusterConstantSinkingPass()); pm.addPass(mlir::TF::CreateResourceDeviceInferencePass()); pm.addNestedPass<FuncOp>( tensorflow::tf2xla::internal::CreateHoistBroadcastReadPass()); pm.addNestedPass<FuncOp>( tensorflow::tf2xla::internal::CreateXlaBroadcastPass());
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/resource_device_inference.cc
} return WalkResult::advance(); }); if (walk_res.wasInterrupted()) return signalPassFailure(); } } } // namespace std::unique_ptr<OperationPass<ModuleOp>> CreateResourceDeviceInferencePass() { return std::make_unique<ResourceDeviceInference>(); } } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/passes.h
const StandardPipelineOptions& options); // Propagates device attributes of resources from callers to callees. std::unique_ptr<OperationPass<ModuleOp>> CreateResourceDeviceInferencePass(); // Creates a pass that promotes resource reads/writes in `functions` to inputs // and outputs of `functions`, assuming that resource operations have already
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
def ResourceDeviceInferencePass : Pass<"tf-resource-device-inference", "ModuleOp"> { let summary = "Propagates the device attribute on resources from callers to " "callees."; let constructor = "TF::CreateResourceDeviceInferencePass()"; let description = [{ A pass that propagates device assignment of resources on a module. It performs in-function propagation, as well as cross-function propagation from callers to callees.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0)