- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 23 for _replication_info (0.28 sec)
-
tensorflow/compiler/mlir/tfrt/tests/runtime_lowering_tpu.mlir
func.func @converts_cluster() { // CHECK: %0:2 = "tf_device.launch"() <{{.*}}> ({ // CHECK: %compilation_status, %program = "tf._TPUCompileMlir"()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 1.6K 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/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/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) -
tensorflow/compiler/mlir/tensorflow/transforms/canonicalize_compile_and_replicate_attributes.cc
// This transformation pass converts existing compilation and replication // attributes into unified attributes. For example, A _tpu_replicate=X // should be replaced with _xla_compile_device_type=TPU and // _replication_info=X attributes by the conversion. An _XlaMustCompile=true // should be replaced with _xla_compile_device_type with the value of device // attribute. #include "llvm/ADT/StringRef.h" #include "llvm/Support/Debug.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 4.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/convert_to_legacy_compile_and_replicate_attributes.cc
==============================================================================*/ // This transformation pass converts unified compilation and replication // attributes into legacy attributes. For example, _replication_info=X // and _xla_compile_device_type=TPU should be replaced with _tpu_replicate=X. // This ensures the unified attributes not get exposed outside of the MLIR // bridge with V1 pipeline in some cases.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 05 23:50:19 UTC 2022 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/attribute_utils.h
"_cluster_outlined_function_name"; // Marks a node for replication. The attribute value indicates the replication // metadata op. inline constexpr llvm::StringRef kReplicationInfoAttr = "_replication_info"; // Marks a node for XLA-TPU compilation. The attribute value indicates the // associated compilation cluster and replication metadata op. inline constexpr llvm::StringRef kTpuReplicateAttr = "_tpu_replicate";
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 19:47:48 UTC 2024 - 8.7K bytes - Viewed (0)