Searched refs:token_instr (Results 1 – 1 of 1) sorted by relevance
1690 HloInstructionProto token_instr; in Infeed() local1691 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Infeed()1692 return AddInstruction(std::move(token_instr), HloOpcode::kAfterAll, {}); in Infeed()1792 HloInstructionProto token_instr; in Outfeed() local1793 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Outfeed()1794 TF_ASSIGN_OR_RETURN(XlaOp token, AddInstruction(std::move(token_instr), in Outfeed()3148 HloInstructionProto token_instr; in Send() local3149 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Send()3150 TF_ASSIGN_OR_RETURN(XlaOp token, AddInstruction(std::move(token_instr), in Send()3191 HloInstructionProto token_instr; in Recv() local[all …]