- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 16 for ClusterFuncOp (0.66 sec)
-
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/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) -
tensorflow/compiler/mlir/tensorflow/transforms/mark_input_output_aliases.cc
assert(aliasing_attr.getInt() == alias_info.output_index); } } void MarkInputOutputAliasesPass::runOnOperation() { SmallVector<tf_device::ClusterFuncOp, 4> cluster_funcs; ModuleOp module = getOperation(); module.walk([&](tf_device::ClusterFuncOp cluster_func) { // Map resource values to pair of input-output indices. llvm::DenseMap<Value, AliasInfo> resource_alias_info_map;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 04:14:26 UTC 2024 - 7.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/parallel_execute_util.h
// Wrap `cluster_func` in a `ParallelExecute` with only one child. This // can be used to canonicalize IR, so there is always one `ParallelExecute`. tf_device::ParallelExecuteOp BuildParallelExecuteOp( tf_device::ClusterFuncOp cluster_func, OpBuilder* builder); // Unwrap `parallel_execute`'s contents if it only has one child. LogicalResult RemoveSingletonParallelExecuteOp( tf_device::ParallelExecuteOp parallel_execute, OpBuilder* builder);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 13 03:57:18 UTC 2023 - 1.7K bytes - Viewed (0)