- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for output_sharding_config (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h
// different outputs depending on the output sharding configuration. mlir::LogicalResult GetOutputTypesForLogicalDeviceComputation( int core_id, llvm::ArrayRef<xla::OpSharding> output_sharding_config, mlir::tf_device::ClusterFuncOp cluster_func, llvm::SmallVectorImpl<mlir::Type>* output_types, llvm::SmallVectorImpl<int>* cluster_to_core_index);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 28 22:18:34 UTC 2024 - 6K bytes - Viewed (0)