- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for XlaClusterFormationPass (0.42 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_cluster_formation.cc
#include "tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h.inc" // Outlines partitioned call ops with `_XlaMustCompile` to device clusters. struct XlaClusterFormationPass : public impl::XlaClusterFormationPassBase<XlaClusterFormationPass> { void runOnOperation() override; }; void EncapsulatePartitionedCall(Operation *call_op, mlir::StringAttr callee_name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 19:09:44 UTC 2023 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
}) : () -> tensor<f32> return %0 : tensor<f32> } ``` }]; let constructor = "tensorflow::tf2xla::internal::CreateExtractOutsideCompilationPass()"; } def XlaClusterFormationPass : Pass<"tf-xla-cluster-formation", "ModuleOp"> { let summary = "Encapsulate partitioned calls within a Cluster op"; let description = [{ This pass clusters `tf.PartitionedCall` and `tf.StatefulPartitionedCall`
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0)