Searched defs:sharded_inputs (Results 1 – 3 of 3) sorted by relevance
573 sharded_inputs, argument817 def generate_enqueue_ops(self, sharded_inputs): argument
3121 absl::InlinedVector<HloInstruction*, 2> sharded_inputs, replicated_inits; in HandleReduceWindow() local
670 std::vector<NodeOut> sharded_inputs; member