- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for ExtractInputsForLogicalDevices (0.45 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
// computation corresponding to i-th logical device. If the attribute does not // exist, the all inputs are placed on logical core 0. mlir::LogicalResult ExtractInputsForLogicalDevices( int num_cores_per_replica, mlir::tf_device::ClusterFuncOp cluster_func, mlir::OpBuilder* builder, llvm::SmallVectorImpl<llvm::SmallVector<mlir::Value, 4>>* input_list);
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/host_runtime/tpu_rewrite_pass.cc
// i-th logical core. llvm::SmallVector<llvm::SmallVector<mlir::Value, 4>, 4> input_list; builder->setInsertionPoint(*new_parallel_execute); auto result = tensorflow::ExtractInputsForLogicalDevices( num_cores_per_replica, cluster_func, builder, &input_list); if (failed(result)) return failure(); const bool replicated = tpu_devices.size() != 1;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
<< hlosharding.status().message(); return; } op->setAttr(shard_str, mlir::StringAttr::get(op->getContext(), hlosharding->ToString())); } mlir::LogicalResult ExtractInputsForLogicalDevices( const int num_cores_per_replica, mlir::tf_device::ClusterFuncOp cluster_func, mlir::OpBuilder* builder, llvm::SmallVectorImpl<llvm::SmallVector<mlir::Value, 4>>* input_list) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K bytes - Viewed (0)