Searched refs:IsFusionComputation (Results 1 – 22 of 22) sorted by relevance
100 if (call_graph_node.computation()->IsFusionComputation()) { in Run()
578 if (!computation->IsFusionComputation()) { in ScheduleModule()601 CHECK(!computation->IsFusionComputation()); in ScheduleComputation()624 if (!computation->IsFusionComputation()) { in Run()
119 CHECK(IsFusionComputation()); in AddParameter()148 CHECK(IsFusionComputation()); in RemoveParameter()170 CHECK(IsFusionComputation()); in RemoveUnusedParameters()205 !IsFusionComputation()) { in IsRemovable()279 if (!IsFusionComputation() && !accept_different_shape) { in set_root_instruction()
128 if (only_fusion_computations_ && !computation->IsFusionComputation()) { in Run()
49 if (computation->IsFusionComputation()) { in Analyze()
394 bool IsFusionComputation() const { return fusion_instruction_ != nullptr; } in IsFusionComputation() function
34 if (computation->IsFusionComputation()) { in FusionCount()
461 if (computation_->IsFusionComputation()) { in Header()561 if (subcomp->IsFusionComputation()) { in ShouldShowSubcomputation()570 if (!subcomp->IsFusionComputation() && MatchTrivialComputation(subcomp)) { in ShouldShowSubcomputation()
635 if (!computation->IsFusionComputation()) { in ComputeSummaryStats()1555 if (computation->IsFusionComputation()) { in BuildColocatedBufferSets()1816 if (computation->IsFusionComputation()) { in CreateAssignment()
455 CHECK(!computation->IsFusionComputation()); in Run()
551 if (c->IsFusionComputation()) { in MakeNonfusionComputations()
340 if (!hlo->parent()->IsFusionComputation()) { in DetermineInstructionPrecision()
1471 if (module->entry_computation()->IsFusionComputation()) { in Run()
1953 if (computation->IsFusionComputation()) { in Run()
1330 CHECK(fused_instructions_computation->IsFusionComputation()) in fused_instructions_computation()
2450 bool HloInstruction::IsFused() const { return parent_->IsFusionComputation(); } in IsFused()
57 if (!computation->IsFusionComputation()) { in ScheduleFromInstructionOrder()
126 !computation->IsFusionComputation()) { in GpuHloOrdering()
298 if (computation->IsFusionComputation()) { in GetIndexTypeForKernel()
243 if (computation->IsFusionComputation()) { in ComputeTargetParallelTasks()
679 if (embedded_computation->IsFusionComputation()) { in RunBackend()867 if (embedded_computation->IsFusionComputation()) { in CompileAheadOfTime()
262 if (comp->IsFusionComputation()) { in DoInfoCommand()