Lines Matching defs:OpFeatures

76 class OpFeatures:  class
233 def register_ephemeral_op(features: OpFeatures):
254 def register_binary_op(features: OpFeatures):
281 def register_unary_op(features: OpFeatures):
292 def register_to_copy_op(features: OpFeatures):
331 def register_mm_op(features: OpFeatures):
348 def register_int8_mm_op(features: OpFeatures):
362 def register_int4_mm_op(features: OpFeatures):
380 def register_softmax_op(features: OpFeatures):
396 def register_reduce_op(features: OpFeatures):
423 def register_2d_pool_op(features: OpFeatures):
437 def register_convolution_op(features: OpFeatures):
450 def register_sdpa_op(features: OpFeatures):
462 def register_rotary_emb_op(features: OpFeatures):
471 def register_view_op(features: OpFeatures):
512 def register_ported_op(features: OpFeatures):
527 def register_ported_ops_with_prepacking(features: OpFeatures):