Searched defs:sharding_for_args (Results 1 – 1 of 1) sorted by relevance
112 const llvm::SmallVectorImpl<llvm::StringRef>& sharding_for_args, in VerifyShardings()185 llvm::SmallVectorImpl<llvm::StringRef>& sharding_for_args) { in IdentifyXlaShardingForComputationInputs()275 const llvm::SmallVectorImpl<llvm::StringRef>& sharding_for_args) { in GetXlaShardingFromAlias()355 const llvm::SmallVectorImpl<llvm::StringRef>& sharding_for_args, in IdentifyXlaShardingForComputationOutputs()428 llvm::SmallVector<llvm::StringRef, 8> sharding_for_args; in IdentifyXlaShardingForTPUComputation() local