Searched refs:DynCast (Results 1 – 13 of 13) sorted by relevance
74 const T* DynCast(const HloInstruction* instruction) { in DynCast() function81 T* DynCast(HloInstruction* instruction) { in DynCast() function83 DynCast<T>(const_cast<const HloInstruction*>(instruction))); in DynCast()92 return instruction != nullptr ? DynCast<T>(instruction) : nullptr; in DynCastOrNull()
76 DynCast<DummyInstruction>(static_cast<HloInstruction*>(&instruction)); in TEST()83 DynCast<DummyInstruction>(static_cast<HloInstruction*>(&instruction)); in TEST()89 ASSERT_DEATH(DynCast<DummyInstruction>(null), ""); in TEST()
100 !DynCast<HloRecvDoneInstruction>(instruction)->is_host_transfer()) { in GlobalPredecessors()106 !DynCast<HloSendInstruction>(instruction)->is_host_transfer()) { in GlobalPredecessors()178 !DynCast<HloRecvInstruction>(instruction)->is_host_transfer()) { in GlobalSuccessors()186 !DynCast<HloSendInstruction>(instruction)->is_host_transfer()) { in GlobalSuccessors()
885 auto send_operand = DynCast<HloSendInstruction>(operand); in CreateSendDone()901 auto recv_operand = DynCast<HloRecvInstruction>(operand); in CreateRecvDone()2880 if (auto hlo_param = DynCast<HloParameterInstruction>(&hlo)) { in ComputeInternal()3177 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in precision_config()3180 if (auto* dot = DynCast<HloDotInstruction>(this)) { in precision_config()3187 if (auto* convolution = DynCast<HloConvolutionInstruction>(this)) { in mutable_precision_config()3190 if (auto* dot = DynCast<HloDotInstruction>(this)) { in mutable_precision_config()3243 auto transpose = DynCast<HloTransposeInstruction>(this); in IsRank2Transpose()3276 return DynCast<HloConstantInstruction>(this) != nullptr; in IsConstant()3351 const HloFusionInstruction* fusion = DynCast<HloFusionInstruction>(this); in IsMultiOutputFusion()[all …]
546 DynCast<const HloCustomCallInstruction>(instruction); in HandleCustomCall()1100 DynCast<const HloSendRecvInstruction>(instr1); in CheckSameIsHostTransfer()1102 DynCast<const HloSendRecvInstruction>(instr2); in CheckSameIsHostTransfer()1121 DynCast<const HloSendRecvInstruction>(instruction); in VerifySendsAndRecvs()
385 DynCast<HloSendRecvInstruction>(instruction); in IsHostSendRecv()395 DynCast<HloSendRecvInstruction>(instruction); in BuildHostChannelConstraints()422 DynCast<HloCustomCallInstruction>(instruction); in IsLayoutConstrainedCustomCall()469 DynCast<HloCustomCallInstruction>(instruction); in AddMandatoryConstraints()675 DynCast<HloCustomCallInstruction>(instruction); in CheckCustomCallLayout()
218 DynCast<HloSendRecvInstruction>(instruction)->is_host_transfer()) { in BytesFreedIfScheduled()
189 DynCast<HloSendRecvInstruction>(instruction); in IsChannelInstruction()
724 return DynCast<HloConstantInstruction>(operand); in TryGetFusionParameterConstant()856 const auto* constant_operand = DynCast<HloConstantInstruction>(operand); in GetInstructionNodeInlinedOperands()
1012 auto* ds = DynCast<HloDynamicSliceInstruction>(user); in CanDoInPlaceDynamicUpdateSlice()
1738 const auto* const_inst = DynCast<HloConstantInstruction>(inst);
256 if (auto call = DynCast<HloCustomCallInstruction>(instr)) { in Run()
996 if (auto* dot_instr = DynCast<HloDotInstruction>(&instr)) { in IsBatchDot()