- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for AddToParallelExecuteOp (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
// `ClusterFunc` with `num_cores_per_replica` children. It keep other children // the same. Return values from the child with the `ClusterFunc` will be // duplicated `num_cores_per_replica` times. LogicalResult AddToParallelExecuteOp( llvm::ArrayRef<llvm::SmallVector<tensorflow::TPUDeviceAndHost, 8>> tpu_devices, llvm::ArrayRef<xla::OpSharding> output_sharding_config,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)