Searched refs:layout_constraint (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | layout_assignment.cc | 1426 const LayoutConstraint* layout_constraint = worklist.front(); in PropagateConstraints() local
1428 VLOG(2) << "Propagating " << layout_constraint->ToString() in PropagateConstraints()
1430 << layout_constraint->priority() << "\n"; in PropagateConstraints()
1432 dynamic_cast<const BufferLayoutConstraint*>(layout_constraint)) { in PropagateConstraints()
1437 layout_constraint)) { in PropagateConstraints()
1442 layout_constraint)) { in PropagateConstraints()
1446 LOG(FATAL) << "Invalid constraint type: " << *layout_constraint; in PropagateConstraints()
1852 const ComputationLayoutConstraint& layout_constraint, in PropagateResultConstraint() argument
1857 layout_constraint.computation_layout().result_layout(), in PropagateResultConstraint()
|
| H A D | layout_assignment.h | 404 const ComputationLayoutConstraint& layout_constraint,
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | graph.py | 934 layout_constraint = None
945 layout_constraint = constrain_fn
946 return layout_constraint, args, kwargs
956 layout_constraint, args, kwargs = get_custom_op_layout_constraints(
968 make_fallback(target, layout_constraint)
|
| H A D | lowering.py | 1778 def make_fallback(op, layout_constraint=None, warn=True): argument
1809 if layout_constraint is not None:
1810 add_layout_constraint(op_overload, layout_constraint)
|