HomeSort by relevance Sort by last modified time
    Searched refs:HloRecvInstruction (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_module_group_util.cc 178 !DynCast<HloRecvInstruction>(instruction)->is_host_transfer()) {
hlo_instructions.h 264 class HloRecvInstruction : public HloSendRecvInstruction {
266 explicit HloRecvInstruction(const Shape& shape, HloInstruction* token,
278 explicit HloRecvDoneInstruction(HloRecvInstruction* operand,
    [all...]
hlo_instructions.cc 436 HloRecvInstruction::HloRecvInstruction(const Shape& shape,
447 std::unique_ptr<HloInstruction> HloRecvInstruction::CloneWithNewOperandsImpl(
451 return absl::make_unique<HloRecvInstruction>(
456 HloRecvDoneInstruction::HloRecvDoneInstruction(HloRecvInstruction* operand,
473 Cast<HloRecvInstruction>(new_operands[0]), is_host_transfer());
    [all...]
hlo_instruction.cc     [all...]

Completed in 322 milliseconds