- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for IdentifyXlaShardingForTPUComputation (0.28 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
return absl::OkStatus(); } // Extracts input/output sharding configuration of `cluster_func` by parsing // XlaSharding ops inside the `cluster_func`. LogicalResult IdentifyXlaShardingForTPUComputation( Builder* builder, mlir::tf_device::ClusterFuncOp cluster_func) { // Look up function definition from module. mlir::func::FuncOp func = cluster_func->getParentOfType<ModuleOp>()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0)