Searched refs:operand_adaptor (Results 1 – 1 of 1) sorted by relevance
540 lmhlo::BroadcastInDimOp::Adaptor operand_adaptor(args); in matchAndRewrite() local541 auto result_type = operand_adaptor.output().getType().cast<MemRefType>(); in matchAndRewrite()564 /*outputBuffers=*/ValueRange{operand_adaptor.output()}, in matchAndRewrite()576 /*outputBuffers=*/ValueRange{operand_adaptor.output()}, indexing_maps, in matchAndRewrite()590 lmhlo::BroadcastInDimOp::Adaptor operand_adaptor(args); in InsertReshapeIfNecessary() local591 Value operand = operand_adaptor.operand(); in InsertReshapeIfNecessary()592 auto operand_type = operand_adaptor.operand().getType().cast<MemRefType>(); in InsertReshapeIfNecessary()595 Value result = operand_adaptor.output(); in InsertReshapeIfNecessary()650 operand_adaptor.operand(), in InsertReshapeIfNecessary()