- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for IdentifyXlaShardingForComputationInputs (0.54 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
// argument to the `cluster_func` directly feeds into another function call op, // then recursively walk the function definition to find the connected // XlaSharding op. void IdentifyXlaShardingForComputationInputs( const llvm::SmallVector<std::string>& logical_device_vec, bool infer_from_computation, mlir::tf_device::ClusterFuncOp cluster_func, mlir::func::FuncOp func, Builder* builder,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0)