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
1618 HloComputation* fused_computation = fusion->fused_instructions_computation(); in CheckFusionInstruction() local
838 auto* fused_computation = fusion->fused_instructions_computation(); in TEST_F() local
361 auto* fused_computation = in CreateFromProto() local
2763 HloComputation* fused_computation = module->entry_computation(); in GetOrCreateSubComputationFromRegion() local3722 const HloComputation* fused_computation = in BuildFusedInitializerThunkForMlir() local4150 const HloComputation* fused_computation = in EmitTileElementForFusion() local4217 HloComputation* fused_computation, FusedIrEmitter* fused_emitter, in EmitPrologueForReduction()4596 absl::Span<const int> instr_index_group, HloComputation* fused_computation, in EmitTileElementForReduction()5397 HloComputation* fused_computation, FusedIrEmitter* fused_emitter, in EmitIRForReduction()5478 HloComputation* fused_computation, int num_reduces) { in DivideOutputInstructionsIntoGroups()5601 HloComputation* fused_computation = nullptr; in EmitReductionFromOrToContiguousDimensions() local
1068 xla::XlaComputation fused_computation; in ExportXlaOp() local
157 const XlaComputation& fused_computation) { in BuildFusion()