Searched refs:token_instr (Results 1 – 1 of 1) sorted by relevance
1309 HloInstructionProto token_instr; in Infeed() local1310 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Infeed()1311 return AddInstruction(std::move(token_instr), HloOpcode::kAfterAll, {}); in Infeed()1409 HloInstructionProto token_instr; in Outfeed() local1410 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Outfeed()1411 TF_ASSIGN_OR_RETURN(XlaOp token, AddInstruction(std::move(token_instr), in Outfeed()2316 HloInstructionProto token_instr; in Send() local2317 *token_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in Send()2318 TF_ASSIGN_OR_RETURN(XlaOp token, AddInstruction(std::move(token_instr), in Send()2359 HloInstructionProto token_instr; in Recv() local[all …]