Searched refs:mlir_arg (Results 1 – 1 of 1) sorted by relevance
528 mlir::BlockArgument mlir_arg = main_fn.getArgument(idx); in RewriteWithArgs() local548 auto tensor_type = mlir_arg.getType().cast<mlir::TensorType>(); in RewriteWithArgs()550 mlir_arg.setType( in RewriteWithArgs()553 mlir_arg.setType(mlir::UnrankedTensorType::get(resource_type)); in RewriteWithArgs()567 mlir_arg.replaceAllUsesWith(constant); in RewriteWithArgs()