Cross Reference: /external/tensorflow/tensorflow/compiler/xla/service/layout

Lines Matching refs:constraints
45 // Abstract base class for layout constraints. These constraint objects are
126 // Class encapsulating the layout constraints of the values in a HLO
140   // Return a vector containing the constraints which have been added to the
143   // identify newly added constraints when propagating layouts.
213   // A vector which holds constraints as they are added. Can be cleared with
217   // Points-to analysis for the module. Used to propagate constraints through
227 // Contains constraints on the layout of channels; sends and recvs.
293   // the cost of the instructions being constrainted on. New constraints are
300       LayoutConstraints* constraints);
303       LayoutConstraints* constraints);
306       LayoutConstraints* constraints);
330       LayoutConstraints* constraints);
337                                       LayoutConstraints* constraints);
355   // Adds constraints which must be satisfied for correctness on all
356   // backends. Called once prior to propagating constraints.
360       HloComputation* computation, LayoutConstraints* constraints);
362   // This method can be overridden to add backend-specific constraints to the
364   // all mandatory constraints have been added via AddMandatoryConstraints
365   // and before propagating constraints.
366   virtual Status AddBackendConstraints(LayoutConstraints* constraints) {
371   // computation satisfying the given ComputationLayout. Layouts constraints are
380   // layout constraints. Copies may be added to satisfy the constraints. The
381   // given LayoutConstraints must have layout constraints every logical buffer
383   Status AssignLayouts(const LayoutConstraints& constraints,
386   // Propagates layout constraints from a set of initial constraints in order to
389   Status PropagateConstraints(LayoutConstraints* constraints);