Home
last modified time | relevance | path

Searched refs:add_layout_constraint (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_inductor/kernel/
H A Dmm_scaled.py10 from ..lowering import add_layout_constraint, constrain_to_fx_strides, register_lowering
241 add_layout_constraint(aten._scaled_mm.default, constrain_to_fx_strides)
H A Dconv.py13 add_layout_constraint,
679 add_layout_constraint(aten.convolution, constrain_conv_to_fx_strides)
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dlowering.py145 def add_layout_constraint(fn, constraint): function
1810 add_layout_constraint(op_overload, layout_constraint)