Searched defs:sharding_for_args (Results 1 – 1 of 1) sorted by relevance
133 llvm::SmallVectorImpl<llvm::StringRef>& sharding_for_args) { in IdentifyXlaShardingForComputationInputs()306 llvm::SmallVector<llvm::StringRef, 8> sharding_for_args; in IdentifyXlaShardingForTPUComputation() local