Home
last modified time | relevance | path

Searched refs:send_done_instr (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2345 HloInstructionProto send_done_instr; in SendWithToken() local
2346 *send_done_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in SendWithToken()
2347 send_done_instr.set_channel_id(handle.handle()); in SendWithToken()
2348 return AddInstruction(std::move(send_done_instr), HloOpcode::kSendDone, in SendWithToken()
2444 HloInstructionProto send_done_instr; in SendToHost() local
2445 *send_done_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in SendToHost()
2446 send_done_instr.set_channel_id(handle.handle()); in SendToHost()
2447 send_done_instr.set_is_host_transfer(true); in SendToHost()
2448 return AddInstruction(std::move(send_done_instr), HloOpcode::kSendDone, in SendToHost()