- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 47 for cluster_func (0.17 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_resource_partitioning.mlir
%1 = "tf.ReadVariableOp"(%0) : (tensor<!tf_type.resource<tensor<i32>>>) -> tensor<i32> // CHECK: [[COMPUTATION:%.+]] = "tf_device.cluster_func"([[INPUT]]) %2 = "tf_device.cluster_func"(%1) {func = @computation, use_spmd_for_xla_partitioning = true} : (tensor<i32>) -> tensor<i32> // CHECK: [[OUTPUT:%.+]]:2 = "tf.TPUPartitionedOutputV2"([[COMPUTATION]]) // CHECK-SAME: _XlaSharding = ""
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 23:53:20 UTC 2024 - 15.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/runtime_passes.td
let summary = "Rewrites a `tf_device.cluster_func` on TPUs into TPU runtime operations."; let description = [{ This pass rewrites a `tf_device.cluster_func` operation into a sequence of `tf._TPUCompileMlir` and `tf.TPUExecute` operations. `tf._TPUCompileMlir` contains a MLIR module that is functionally equivalent to the function referenced by `tf_device.cluster_func`.
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/tests/mark_input_output_aliases.mlir
// RUN: tf-opt %s -tf-device-mark-input-output-aliases | FileCheck %s // The following tests check if the aliasing pairs are conservatively marked // correctly. In the following tests tf_device.cluster_func has inputs // coming from ReadVariableOp and outputs written to a resource using // AssignVariableOp. If a pair of input-output (say input at index `a` and // output at index `b`) read and write to the same resource, then those // input-output pairs alias each other.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 04:14:26 UTC 2024 - 6.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/xla_rewrite.mlir
%0 = "tf_device.cluster_func"(%arg0, %arg1) {func = @func_with_resources} : (tensor<!tf_type.resource>, tensor<i32>) -> tensor<i32> // CHECK: "tf.XlaLaunch"(%arg1, %arg0) <{function = @func_with_resources, operandSegmentSizes = array<i32: 0, 1, 1>}> : (tensor<i32>, tensor<!tf_type.resource>) -> tensor<i32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir
// Tests `tf_device.cluster_func` with missing `step_marker_location` attribute. module attributes {tf.versions = {producer = 888 : i32}, tf.devices = ["/job:worker/replica:0/task:0/device:CPU:0", "/job:worker/replica:0/task:0/device:TPU_SYSTEM:0", "/job:worker/replica:0/task:0/device:TPU:0"]} { func.func @bad_num_cores_per_replica() { // expected-error@+1 {{requires attribute 'step_marker_location'}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
BuildSpaceToDepth(cluster_func, input, block_size, input_shape); cluster_func.setOperand(index, space_to_depth); return space_to_depth; } // Performs transformation for replicated inputs. Returns true if this is a // supported case (thus transform happened). bool HandleHostReplicatedInputs(int64_t index, tf_device::ClusterFuncOp cluster_func,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 29.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util_test.cc
module->walk([&](mlir::tf_device::ClusterFuncOp cluster_func) { cluster_func_ops.push_back(cluster_func); }); EXPECT_EQ(cluster_func_ops.size(), 1); EXPECT_TRUE(mlir::succeeded(tensorflow::EraseClusterFuncs(cluster_func_ops))); llvm::SmallVector<mlir::tf_device::ClusterFuncOp, 4> new_cluster_func_ops; module->walk([&](mlir::tf_device::ClusterFuncOp cluster_func) { new_cluster_func_ops.push_back(cluster_func); });
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/annotate_parameter_replication.cc
} return v; } void AnnotateParameterReplicationPass::runOnOperation() { ModuleOp m = getOperation(); OpBuilder builder(m.getContext()); m.walk([&](tf_device::ClusterFuncOp cluster_func) { auto replicate = cluster_func->getParentOfType<tf_device::ReplicateOp>(); if (!replicate) return; auto mirrored_variable_indices_attr = replicate->getAttrOfType<ArrayAttr>(kMirroredVariableIndicesAttr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
mlir::OpBuilder* builder, llvm::SmallVectorImpl<llvm::SmallVector<mlir::Value, 4>>* input_list); // Extracts a list of OpSharding that represent output sharding configuration of // `tf_device.cluster`. mlir::LogicalResult ParseAndValidateOutputSharding( int num_cores_per_replica, mlir::tf_device::ClusterFuncOp cluster_func,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/rewrite_cluster_to_ifrt_call.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 17 07:28:40 UTC 2024 - 9K bytes - Viewed (0)