Searched defs:token_operand (Results 1 – 2 of 2) sorted by relevance
1383 const Shape& infeed_shape, HloInstruction* token_operand, in CreateInfeed()1391 HloInstruction* token_operand, absl::string_view outfeed_config) { in CreateOutfeed()1452 HloInstruction* token_operand) { in CreateAddDependency()
2346 HloInstruction* token_operand, in HloInfeedInstruction()2387 HloInstruction* token_operand, in HloOutfeedInstruction()