- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 32 for _replication_info (0.33 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
return success(); } // Forms clusters with ops of the same `_replication_info` attribute under a // block. // // For a given block, clusters are formed via grouping ops by // `_replication_info` attributes. For every cluster formed: // 1. Find associated TPUReplicateMetadata attributes with the same // `_replication_info` attribute. // 2. Find users not in cluster that are interleaved between cluster ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/canonicalize_compile_and_replicate_attributes.mlir
%control = tf_executor.island wraps "tf.TPUReplicateMetadata"() {_tpu_replicate = "cluster", allow_soft_placement = false, computation_shape...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 3.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/executor_tpuv1_island_coarsening.cc
// already have a cluster assigned to them (via `_replication_info` // attribute value). // `tf.Identity` op is also treated as special tpu ops since it can play // a role as connection between `tf.TPUReplicatedInput` or // `tf.TPUPartitionedInput`. For example, we have the follow pseudocode: // %0 = tf_executor.island wraps "tf.OpA" (){_replication_info = 'c'}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 27.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/runtime_passes.td
For example, a non replicated `tf_device.cluster_func`: ```mlir func @tf_tpu_rewrite(%arg0: tensor<i8>) { %0 = "tf_device.cluster_func"(%arg0) {_xla_compile_device_type = "TPU", _replication_info = "cluster0", func = @func} : (tensor<i8>) -> tensor<i8> return } ``` will be rewritten as: ```mlir func @tf_tpu_rewrite(%arg0: tensor<i8>) {
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/testdata/basic_cluster.mlir
func.func @main() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 17 16:43:16 UTC 2023 - 639 bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/runtime_lowering_gpu.mlir
// CHECK-LABEL: @converts_cluster func.func @converts_cluster() { // CHECK: "tf.XlaLaunch"()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Oct 13 17:41:44 UTC 2023 - 840 bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/testdata/spmd.mlir
func.func @main(%arg0: tensor<*xf32> {tf.device = "/job:localhost/replica:0/task:0/device:CPU:0"}) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 12 04:22:33 UTC 2023 - 1.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Aug 02 02:26:39 UTC 2023 - 96.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.h
// Verifies that all MLIR Ops have the expected attributes. std::unique_ptr<mlir::OperationPass<mlir::func::FuncOp>> CreateVerifyClusteringPass(); // Creates a pass that forms clusters from operations of the same // `_replication_info` attribute. std::unique_ptr<mlir::OperationPass<mlir::ModuleOp>> CreateTPUClusterFormationPass(bool strict_clusters = false); // Creates a pass that extracts outside compilation (Host ops inside device
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu-annotate-dynamic-shape-inputs.mlir
tf_device.return %4#0, %4#1 : tensor<2048xi32>, tensor<2048xi32> }) {device = "TPU_REPLICATED_HOST_0"} : () -> (tensor<2048xi32>, tensor<2048xi32>) %1 = "tf_device.cluster_func"(%0#0, %0#1) {_replication_info = "cluster_test_fn", func = @tpu_func} : (tensor<2048xi32>, tensor<2048xi32>) -> tensor<2048xi32> return %1: tensor<2048xi32> } // CHECK-LABEL: func @tpu_func // CHECK: mhlo.type_extensions
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Aug 14 15:35:49 UTC 2023 - 2.3K bytes - Viewed (0)