- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 24 for ClusterFuncOp (0.34 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.cc
return llvm::formatv("requires attribute '{0}'", attribute).str(); } // Populates a TPUCompileMetadataProto with StepMarkerLocation from a // `tf_device::ClusterFuncOp`. LogicalResult SetMetadataProtoStepMarkerLocation( tf_device::ClusterFuncOp op, tensorflow::tpu::TPUCompileMetadataProto* metadata) { auto step_marker_location = op->getAttrOfType<StringAttr>(kStepMarkerLocationAttr);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils_test.cc
context_.loadAllAvailableDialects(); } absl::StatusOr<std::vector<mlir::tf_device::ClusterFuncOp>> GetClusterFuncOps( absl::string_view mlir_module_filename) { TF_RETURN_IF_ERROR(CreateMlirModule(mlir_module_filename)); std::vector<mlir::tf_device::ClusterFuncOp> cluster_func_ops; mlir_module_->walk([&](mlir::tf_device::ClusterFuncOp op) { cluster_func_ops.push_back(op); }); return cluster_func_ops; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 12 04:22:33 UTC 2023 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.h
#include "tensorflow/core/util/device_name_utils.h" namespace tensorflow { // Erase rewritten ClusterFuncOp(s). If TPUPartitionedInputV2Op / // TPUPartitionedOutputV2Op are present, they must be removed along with the // ClusterFuncOp(s). mlir::LogicalResult EraseClusterFuncs( llvm::MutableArrayRef<mlir::tf_device::ClusterFuncOp> to_be_erased); // Move child processes of the ParallelExecute that do not change. These are all
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_read_for_write.cc
SmallVector<tf_device::ClusterFuncOp, 4> cluster_funcs; getOperation().walk([&](tf_device::ClusterFuncOp cluster_func) { cluster_funcs.push_back(cluster_func); }); OpBuilder builder(&getContext()); // Add resource reads for resource writes from TPU cluster where for such // resources the TPU cluster does not read from. for (tf_device::ClusterFuncOp cluster_func : cluster_funcs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 16:54:40 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_resource_partitioning.cc
} void TPUResourceReadsWritesPartitioningPass::runOnOperation() { llvm::SmallVector<tf_device::ClusterFuncOp, 4> cluster_funcs; getOperation()->walk([&cluster_funcs](tf_device::ClusterFuncOp cluster_func) { cluster_funcs.push_back(cluster_func); }); for (tf_device::ClusterFuncOp cluster_func : cluster_funcs) if (failed(PartitionResourceReadsWrites(cluster_func))) return signalPassFailure(); }
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/tensorflow/utils/xla_rewrite_util_test.cc
llvm::SmallVector<mlir::tf_device::ClusterFuncOp, 4> cluster_func_ops; 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;
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/utils/xla_rewrite_util.cc
#include "tensorflow/compiler/mlir/tensorflow/utils/xla_rewrite_util.h" namespace tensorflow { mlir::LogicalResult EraseClusterFuncs( llvm::MutableArrayRef<mlir::tf_device::ClusterFuncOp> to_be_erased) { for (auto cluster : to_be_erased) { auto old_parallel_execute = cluster->getParentOfType<mlir::tf_device::ParallelExecuteOp>(); if (!old_parallel_execute) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 4.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.h
namespace mlir { namespace TFTPU { // Populates a TPUCompileMetadataProto from attributes of a // `tf_device::ClusterFuncOp`. If any necessary attributes are missing from the // op, a failure will be returned. // TODO(lyandy): Support session handle and guaranteed consts. LogicalResult SetMetadataProtoFromClusterFuncOp( tf_device::ClusterFuncOp op, int num_replicas, int num_cores_per_replica,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 12 04:22:33 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
int num_cores_per_replica, mlir::tf_device::ClusterFuncOp cluster_func, 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/tensorflow/transforms/xla_rewrite.cc
callee.getBody().getArgumentTypes(), callee.getResultTypes())); } void RewriteCall(tf_device::ClusterFuncOp cluster_func_op, SymbolTable &symtab, OpBuilder &builder) { llvm::SmallVector<Value> non_resource_args, resource_args; bool has_resources = false, in_order = true;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0)