Searched defs:LinalgStructuredOpConfig (Results 1 – 3 of 3) sorted by relevance
39 def prepare_common_structured_op(op_config: LinalgStructuredOpConfig,170 def emit_generic_structured_op(op_config: LinalgStructuredOpConfig, *ins: Value,225 def emit_named_structured_op(op_config: LinalgStructuredOpConfig, op_name: str,475 op_config: LinalgStructuredOpConfig,
112 class LinalgStructuredOpConfig(YAMLObject): class415 structured_op: Optional[LinalgStructuredOpConfig] = None):
121 struct LinalgStructuredOpConfig { struct122 SmallVector<LinalgOperandDef> args;123 LinalgIndexingMapsConfig indexingMaps;124 SmallVector<LinalgIteratorTypeDef> iteratorTypes;125 std::vector<ScalarAssign> assignments;