Lines Matching defs:state

50                                     transform::TransformState &state) {
62 static ParseResult parse(OpAsmParser &parser, OperationState &state) {
69 state.addAttribute("message", message);
106 transform::TransformState &state) {
118 transform::TransformResults &results, transform::TransformState &state) {
138 transform::TransformResults &results, transform::TransformState &state) {
154 transform::TransformState &state) {
172 transform::TransformState &state) {
196 transform::TransformState &state) {
210 transform::TransformResults &results, transform::TransformState &state) {
211 auto payload = state.getPayloadOps(getOperand());
233 transform::TransformState &state) {
251 transform::TransformResults &results, transform::TransformState &state) {
252 state.addExtension<TestTransformStateExtension>(getMessageAttr());
259 transform::TransformResults &results, transform::TransformState &state) {
260 auto *extension = state.getExtension<TestTransformStateExtension>();
268 for (Operation *payload : state.getPayloadOps(getOperand())) {
272 assert(succeeded(state.getHandlesForPayloadOp(payload, handles)));
290 transform::TransformResults &results, transform::TransformState &state) {
291 auto *extension = state.getExtension<TestTransformStateExtension>();
296 *state.getPayloadOps(getOperand()).begin(), getOperation())))
312 transform::TransformResults &results, transform::TransformState &state) {
313 state.removeExtension<TestTransformStateExtension>();
319 transform::TransformResults &results, transform::TransformState &state) {
320 auto payloadOps = state.getPayloadOps(getTarget());
328 transform::TransformResults &results, transform::TransformState &state) {
338 transform::TransformResults &results, transform::TransformState &state) {
347 transform::TransformResults &results, transform::TransformState &state) {
349 for (Operation *op : state.getPayloadOps(getTarget()))
366 transform::TransformState &state) {
376 transform::TransformState &state) {
389 transform::TransformState &state) {
400 transform::TransformState &state) {
411 transform::TransformState &state) {
420 transform::TransformState &state) {
422 state.getPayloadOps(getHandle()));
469 transform::TransformResults &results, transform::TransformState &state) {
471 for (Operation *op : state.getPayloadOps(getTarget())) {
474 (void)state.getHandlesForPayloadOp(nested, handles);
485 transform::TransformState &state) {
489 llvm::map_range(state.getParams(param), [](Attribute attr) -> uint32_t {
507 transform::TransformResults &results, transform::TransformState &state) {
510 llvm::map_range(state.getPayloadOps(getHandle()),
526 transform::TransformState &state) {
541 ::transform::TransformState &state) {
548 results.push_back(*state.getPayloadOps(getIn()).begin());
552 results.push_back(*state.getPayloadOps(getIn()).begin());
567 transform::TransformResults &results, transform::TransformState &state) {
575 transform::TransformResults &results, transform::TransformState &state) {
587 transform::TransformResults &results, transform::TransformState &state) {
599 transform::TransformResults &results, transform::TransformState &state) {
623 transform::TransformResults &results, transform::TransformState &state) {
624 results.set(llvm::cast<OpResult>(getOut()), state.getPayloadOps(getIn()));
648 transform::TransformState &state) {
653 for (Operation *op : state.getPayloadOps(getIn())) {
658 for (Operation *op : state.getPayloadOps(getIn())) {
730 transform::TransformState &state) {
735 state.getPayloadOps(getOperand(arg.getArgNumber())),
740 auto scope = state.make_region_scope(getBody());
742 if (failed(state.mapBlockArgument(arg, mappings[arg.getArgNumber()])))
747 state.applyTransform(cast<transform::TransformOpInterface>(op));
764 transform::TransformResults &results, transform::TransformState &state) {
765 auto originalOps = state.getPayloadOps(getOriginal());
766 auto replacementOps = state.getPayloadOps(getReplacement());
793 transform::TransformState &state) {
809 transform::TransformResults &results, transform::TransformState &state) {
813 state.getExtension<TransformStateInitializerExtension>();
817 state.addExtension<TransformStateInitializerExtension>(1, opCollection);