Searched defs:token_operand (Results 1 – 2 of 2) sorted by relevance
1286 const Shape& infeed_shape, HloInstruction* token_operand, in CreateInfeed()1294 HloInstruction* token_operand, absl::string_view outfeed_config) { in CreateOutfeed()1355 HloInstruction* token_operand) { in CreateAddDependency()
2186 HloInstruction* token_operand, in HloInfeedInstruction()2227 HloInstruction* token_operand, in HloOutfeedInstruction()