- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for ExtractInputsForLogicalDevices (0.3 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)