Searched refs:fusion_operand (Results 1 – 5 of 5) sorted by relevance
121 HloInstruction* fusion_operand) { in RemoveFusionOperand() argument123 std::move(indexing_users_.at(fusion_operand)); in RemoveFusionOperand()124 indexing_users_.erase(fusion_operand); in RemoveFusionOperand()125 CHECK(!index_usage_count_.contains(fusion_operand)); in RemoveFusionOperand()
92 const HloInstruction* fusion_operand = in Propagate() local94 Propagate(position.index, fusion_operand, memory_space); in Propagate()
59 HloInstruction* fusion_operand);
1685 const HloInstruction* fusion_operand = in SetFusionLayouts() local1687 DCHECK(ShapeUtil::Compatible(fusion_operand->shape(), in SetFusionLayouts()1690 fusion_operand->shape(), fused_instruction->mutable_shape())); in SetFusionLayouts()
76 HloInstruction* fusion_operand; in CanEmitFusedDynamicUpdateSliceInPlace() local78 std::tie(fusion_operand, index) = in CanEmitFusedDynamicUpdateSliceInPlace()82 CHECK_EQ(fusion_operand->opcode(), HloOpcode::kParameter); in CanEmitFusedDynamicUpdateSliceInPlace()83 auto* operand = fusion->operand(fusion_operand->parameter_number()); in CanEmitFusedDynamicUpdateSliceInPlace()