Searched refs:ComputationLayoutConstraint (Results 1 – 2 of 2) sorted by relevance
130 class ComputationLayoutConstraint : public LayoutConstraint {136 explicit ComputationLayoutConstraint(const HloComputation* computation, in ComputationLayoutConstraint() function290 const ComputationLayoutConstraint& computation_constraint() const { in computation_constraint()293 ComputationLayoutConstraint* mutable_computation_constraint() { in mutable_computation_constraint()304 ComputationLayoutConstraint computation_constraint_;404 const ComputationLayoutConstraint& layout_constraint,
125 std::string ComputationLayoutConstraint::ToString() const { in ToString()689 ComputationLayoutConstraint* body_constraint = in AddMandatoryConstraints()693 ComputationLayoutConstraint* condition_constraint = in AddMandatoryConstraints()1441 dynamic_cast<const ComputationLayoutConstraint*>( in PropagateConstraints()1852 const ComputationLayoutConstraint& layout_constraint, in PropagateResultConstraint()2107 ComputationLayoutConstraint* callee_constraint = in CalculateComputationLayout()