Searched refs:IsEntryComputation (Results 1 – 7 of 7) sorted by relevance
74 while (!computation->IsEntryComputation()) { in IsNotContainedInLoop()
456 return (computation_->IsEntryComputation() || in ResultLayout()604 (constraints->computation()->IsEntryComputation() && in AddMandatoryConstraints()802 (constraints->computation()->IsEntryComputation() && in AddMandatoryConstraints()2195 if (constraints->computation()->IsEntryComputation()) { in CalculateComputationLayout()
565 bool IsEntryComputation() const;
1336 bool HloComputation::IsEntryComputation() const { in IsEntryComputation() function in xla::HloComputation
314 instr->parent()->IsEntryComputation()) { in GetOutputInstruction()
359 while (!computation->IsEntryComputation()) { in CalculateComputationNestLevel()
1544 TF_RET_CHECK(computation_.IsEntryComputation()); in Initialize()1571 if (computation_.IsEntryComputation()) { in Initialize()