Searched defs:fused_computation (Results 1 – 12 of 12) sorted by relevance
26 FusedComputationType* fused_computation, in InitializeFusedComputation()
58 FusedComputationType fused_computation; member
47 HloComputation* fused_computation = in Defuse() local
142 HloComputation* fused_computation; in TEST_F() local
485 auto fused_computation = in TEST_F() local
2190 HloComputation* fused_computation = fusion->fused_instructions_computation(); in CheckFusionInstruction() local
937 auto* fused_computation = fusion->fused_instructions_computation(); in TEST_F() local
418 auto* fused_computation = in CreateFromProto() local
755 const HloComputation* fused_computation, std::vector<int64_t> input_ids) { in FilterInputsForShmemTranspose()813 const HloComputation* fused_computation) { in Match021Transpose()
441 const HloComputation* fused_computation) { in GetConsistentInputShapeForRootSlices()2474 HloComputation* fused_computation = module->entry_computation(); in GetOrCreateSubComputationFromRegion() local3273 const HloComputation* fused_computation = in BuildFusedInitializerThunk() local3520 const HloComputation* fused_computation = in EmitTileElementForTranspose() local4842 HloComputation* fused_computation) { in GroupDisjointReductions()5036 const HloComputation* fused_computation, in EmitElementForInputFusibleSlices()5149 mlir::lmhlo::FusionOp fusion_op, const HloComputation* fused_computation) { in EmitDynamicUpdateSlice()5196 const HloComputation* fused_computation) { in EmitScatter()
1691 xla::XlaComputation fused_computation; in ExportXlaOp() local
115 const XlaComputation& fused_computation) { in BuildFusion()