- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for get_XlaShardingAttr (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
llvm::SmallVector<Value, 4> new_reads, bool is_packed) { xla::OpSharding sharding; sharding.ParseFromString( old_partitioned_input.get_XlaShardingAttr().getValue().str()); for (OpOperand& read_use : llvm::make_early_inc_range(old_read.getValue().getUses())) { if (dyn_cast_or_null<tf_device::ClusterFuncOp>(read_use.getOwner())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
// Assume all the replicas have the same structure. mlir::TF::TPUPartitionedOutputV2Op first_op = *(ops.begin()); mlir::ArrayAttr dims = first_op.getPartitionDimsAttr(); StringAttr sharding = first_op.get_XlaShardingAttr(); Operation::result_type_range output_types = first_op.getResultTypes(); result_op = builder.create<mlir::TF::TPUPartitionedOutputV2Op>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0)