Searched defs:fused_computation (Results 1 – 11 of 11) sorted by relevance
26 FusedComputationType* fused_computation, in InitializeFusedComputation()
57 FusedComputationType fused_computation; member
48 HloComputation* fused_computation = in Defuse() local
118 HloComputation* fused_computation; in TEST_F() local
507 auto fused_computation = in TEST_F() local
1970 HloComputation* fused_computation = fusion->fused_instructions_computation(); in CheckFusionInstruction() local
838 auto* fused_computation = fusion->fused_instructions_computation(); in TEST_F() local
362 auto* fused_computation = in CreateFromProto() local
467 const HloComputation* fused_computation) { in GetConsistentInputShapeForRootSlices()2613 HloComputation* fused_computation = module->entry_computation(); in GetOrCreateSubComputationFromRegion() local3406 const HloComputation* fused_computation = in BuildFusedInitializerThunk() local3662 const HloComputation* fused_computation = in EmitTileElementForFusion() local3723 HloComputation* fused_computation, FusedIrEmitter* fused_emitter, in EmitPrologueForReduction()4154 absl::Span<const int> instr_index_group, HloComputation* fused_computation, in EmitTileElementForReduction()4925 HloComputation* fused_computation, FusedIrEmitter* fused_emitter, in EmitIRForReduction()5001 HloComputation* fused_computation, int num_reduces) { in GroupDisjointReductions()5092 HloComputation* fused_computation = nullptr; in EmitUnnestedReduction() local5227 const HloComputation* fused_computation, in EmitElementForInputFusibleSlices()
1193 xla::XlaComputation fused_computation; in ExportXlaOp() local
103 const XlaComputation& fused_computation) { in BuildFusion()