Searched defs:token_operand (Results 1 – 2 of 2) sorted by relevance
1109 const Shape& infeed_shape, HloInstruction* token_operand, in CreateInfeed()1117 HloInstruction* token_operand, absl::string_view outfeed_config) { in CreateOutfeed()1178 HloInstruction* token_operand) { in CreateAddDependency()
2048 HloInstruction* token_operand, in HloInfeedInstruction()2089 HloInstruction* token_operand, in HloOutfeedInstruction()