Searched refs:outfeed_token (Results 1 – 2 of 2) sorted by relevance
2116 HloInstruction* const outfeed_token = operands[1]; in ParseInstructionRhs() local2120 shape, outfeed_input, outfeed_token, config ? *config : "")); in ParseInstructionRhs()
2979 auto outfeed_token = branch_b.AddInstruction( in HandleOutfeed() local3029 per_branch_partitioned_shapes[i], outfeed_data, outfeed_token, in HandleOutfeed()