Searched defs:OpFeatures (Results 1 – 3 of 3) sorted by relevance
76 class OpFeatures: class233 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):[all …]
69 self, node: torch.fx.Node, features: Optional[OpFeatures] = None139 self, node: torch.fx.Node, features: Optional[OpFeatures] = None
19 class OpFeatures: class